记录编号 307695 评测结果 AAAAAAAAAA
题目名称 [ZJOI 2008]树的统计Count 最终得分 100
用户昵称 Gravatar521 是否通过 通过
代码语言 C++ 运行时间 1.707 s
提交时间 2016-09-16 08:29:27 内存使用 2.50 MiB
显示代码纯文本
#include<cstdio>
using namespace std;
inline void read(int &x)
{
	char ch;bool flag=false;
	while(ch=getchar(),ch<'0'||ch>'9')if(ch=='-')flag=1;
	x=ch^'0';
	while(ch=getchar(),ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^'0');
	if(flag) x=-x;
}
//===============================================================
const int Size=30010,INF=-0x7ffffff;
int son[Size],top[Size],dep[Size],tid[Size],rank[Size],fa[Size],siz[Size],tn;
int to[Size<<1],next[Size<<1],head[Size],val[Size],num;
struct Tree{
	int max,sum;
	Tree(){max=INF,sum=0;}
}tree[Size<<2];
inline void Add(int u,int v)
{
	to[++num]=v,next[num]=head[u],head[u]=num;
}
void dfs1(int u,int f,int d)
{
	son[u]=0,siz[u]=1,fa[u]=f,dep[u]=d;
	for(int i=head[u];i;i=next[i]){
		int v=to[i];
		if(v==f) continue;
		dfs1(v,u,d+1);
		siz[u]+=siz[v];
		if(siz[son[u]]<siz[v]) son[u]=v;
	}
}
void dfs2(int u,int tp)
{
	top[u]=tp,tid[u]=++tn,rank[tid[u]]=u;
	if(son[u]) dfs2(son[u],tp);
	for(int i=head[u];i;i=next[i])
		if(to[i]!=son[u]&&to[i]!=fa[u])
			dfs2(to[i],to[i]);
}
inline int Max(int x,int y){return x>y?x:y;}
inline void pushup(int rt)
{
	tree[rt].max=Max(tree[rt<<1].max,tree[rt<<1|1].max);
	tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum;
}
void build(int l,int r,int rt)
{
	if(l>r) return ;
	if(l==r){
		tree[rt].max=tree[rt].sum=val[rank[l]];
		return ;
	}
	int mid=l+r>>1;
	build(l,mid,rt<<1);
	build(mid+1,r,rt<<1|1);
	pushup(rt);
}
void update(int p,int x,int l,int r,int rt)
{
	if(l>r) return ;
	if(l==r){
		tree[rt].max=tree[rt].sum=x;
		return ;
	}
	int mid=l+r>>1;
	if(mid>=p) update(p,x,l,mid,rt<<1);
	else update(p,x,mid+1,r,rt<<1|1);
	pushup(rt);
}
int query_max(int L,int R,int l,int r,int rt)
{
	if(L>R||l>r) return INF;
	if(l>=L&&r<=R) return tree[rt].max;
	int mid=r+l>>1,ret=INF;
	if(mid>=L) ret=Max(ret,query_max(L,R,l,mid,rt<<1));
	if(mid<R) ret=Max(ret,query_max(L,R,mid+1,r,rt<<1|1));
	return ret;
}
int query_sum(int L,int R,int l,int r,int rt)
{
	if(L>R||l>r) return 0;
	if(l>=L&&r<=R) return tree[rt].sum;
	int mid=r+l>>1,ret=0;
	if(mid>=L) ret+=query_sum(L,R,l,mid,rt<<1);
	if(mid<R) ret+=query_sum(L,R,mid+1,r,rt<<1|1);
	return ret;
}
inline void swap(int& x,int& y){x^=y,y^=x,x^=y;}
int findmax(int u,int v)
{
	int ret=INF;
	while(top[u]!=top[v]){
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		ret=Max(ret,query_max(tid[top[u]],tid[u],1,tn,1));
		u=fa[top[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	ret=Max(ret,query_max(tid[u],tid[v],1,tn,1));
	return ret;
}
int findsum(int u,int v)
{
	int ret=0;
	while(top[u]!=top[v]){
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		ret+=query_sum(tid[top[u]],tid[u],1,tn,1);
		u=fa[top[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	ret+=query_sum(tid[u],tid[v],1,tn,1);
	return ret;
}
int _521()
{
#define COGS
#ifdef COGS
	freopen("bzoj_1036.in","r",stdin);
	freopen("bzoj_1036.out","w",stdout);
#else
	freopen("1.in","r",stdin);
#endif
	int n,i,q;
	read(n);
	for(i=1;i<n;i++){
		int u,v;
		read(u),read(v);
		Add(u,v),Add(v,u);
	}
	for(i=1;i<=n;i++) read(val[i]);
	dfs1(1,0,1),dfs2(1,1);
	build(1,tn,1);
	read(q);
	for(i=1;i<=q;i++){
		char ch[10];
		int a,b;
		scanf("%s",ch),read(a),read(b);
		if(ch[0]=='Q'){
			if(ch[1]=='M') printf("%d\n",findmax(a,b));
			else printf("%d\n",findsum(a,b));
		}
		else update(tid[a],b,1,tn,1);
	}
	return 0;
}
int _520=_521();
int main(){;}