BZOJ 1036: [ZJOI2008]树的统计Count(树链剖分)

1036: [ZJOI2008]树的统计Count

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 22104  Solved: 8996
[Submit][Status][Discuss]

Description

　　一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成
一些操作： I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I
II. QSUM u v: 询问从点u到点v的路径上的节点的权值和 注意：从点u到点v的路径上的节点包括u和v本身

Input

　　输入的第一行为一个整数n，表示节点的个数。接下来n – 1行，每行2个整数a和b，表示节点a和节点b之间有
一条边相连。接下来n行，每行一个整数，第i行的整数wi表示节点i的权值。接下来1行，为一个整数q，表示操作
的总数。接下来q行，每行一个操作，以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。 
对于100％的数据，保证1<=n<=30000，0<=q<=200000；中途操作中保证每个节点的权值w在-30000到30000之间。

Output

　　对于每个“QMAX”或者“QSUM”的操作，每行输出一个整数表示要求输出的结果。

#include<bits/stdc++.h>
using namespace std;
const int maxn=30005;
int seg[maxn],rev[maxn],size[maxn],son[maxn],top[maxn],dep[maxn];
int sum[maxn<<2],Max[maxn<<2],fa[maxn],a[maxn];
int head[maxn],n,tol,MMax,MMsum;
struct node
{
	int to,next;
}rode[maxn*2];
void add(int a,int b)
{
	rode[tol].to=b;
	rode[tol].next=head[a];
	head[a]=tol++;
}
void dfs1(int v,int father)
{
	size[v]=1;
	fa[v]=father;
	dep[v]=dep[father]+1;
	for(int i=head[v];i!=-1;i=rode[i].next)
	{
		node e=rode[i];
		if(e.to==father) continue;
		dfs1(e.to,v);
		size[v]+=size[e.to];
		if(size[e.to]>size[son[v]]) son[v]=e.to;
	}
}
void dfs2(int v,int father)
{
	if(son[v])
	{
		seg[son[v]]=++seg[0];
		top[son[v]]=top[v];
		rev[seg[0]]=son[v];
		dfs2(son[v],father);
	}
	for(int i=head[v];i!=-1;i=rode[i].next)
	{
		node e=rode[i];
		if(!top[e.to])
		{
			seg[e.to]=++seg[0];
			top[e.to]=e.to;
			rev[seg[0]]=e.to;
			dfs2(e.to,v);
		}
	}
}
void build(int l,int r,int rt)
{
	if(l==r)
	{
		sum[rt]=Max[rt]=a[rev[l]];
		return;
	}
	int mid=(l+r)/2;
	build(l,mid,rt<<1);
	build(mid+1,r,rt<<1|1);
	Max[rt]=max(Max[rt<<1],Max[rt<<1|1]);
	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void update(int L,int C,int l,int r,int rt)
{
	if(l==r)
	{
		sum[rt]=Max[rt]=C;
		return;
	}
	int mid=(l+r)/2;
	if(L<=mid) update(L,C,l,mid,rt<<1);
	else update(L,C,mid+1,r,rt<<1|1);
	Max[rt]=max(Max[rt<<1],Max[rt<<1|1]);
	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void query(int L,int R,int l,int r,int rt)
{
	if(l>=L&&r<=R)
	{
		MMax=max(MMax,Max[rt]);
		MMsum+=sum[rt];
		return;
	}
	int mid=(l+r)/2;
	if(L<=mid) query(L,R,l,mid,rt<<1);
	if(R>mid) query(L,R,mid+1,r,rt<<1|1);
}
void work(int x,int y)
{
	int fx=top[x],fy=top[y];
	while(fx!=fy)
	{
		if(dep[fx]<dep[fy]) swap(x,y),swap(fx,fy);
		query(seg[fx],seg[x],1,seg[0],1);
		x=fa[fx],fx=top[x];
	}
	if(dep[x]>dep[y]) swap(x,y);
	query(seg[x],seg[y],1,seg[0],1);
}
int main()
{
	memset(head,-1,sizeof(head));
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
		int x,y;scanf("%d%d",&x,&y);
		add(x,y);add(y,x);
	}
	for(int i=1;i<=n;i++)
	scanf("%d",&a[i]);
	dfs1(1,0);
	seg[0]=seg[1]=rev[1]=top[1]=1;
	dfs2(1,0);
	build(1,seg[0],1);
	int Q;scanf("%d",&Q);
	while(Q--)
	{
		char t[10];int x,y;
		scanf("%s%d%d",&t,&x,&y);
		if(t[0]=='C') update(seg[x],y,1,seg[0],1);
		else 
		{
			MMax=-1e9,MMsum=0;
			work(x,y);
			if(t[1]=='M') printf("%d\n",MMax);
			else printf("%d\n",MMsum);
		}
	}
	return 0;
}