树的统计 洛谷2590 BZOJ1036 zjoi2008

题意：

一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。

我们将以下面的形式来要求你对这棵树完成一些操作：
I. CHANGE u t : 把结点u的权值改为t
II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值
III. QSUM u v: 询问从点u到点v的路径上的节点的权值和
注意：从点u到点v的路径上的节点包括u和v本身
一道树剖模板，然而我却因为一些细节写挂了好多次。。。
本题需注意的点：第一，修改的时候是单点修改，所以没有必要弄标记也没必要写传标记的函数；第二，本题有负数，所以在查询最大值时要把答案初始值赋为负无穷；第三，本题查询时传入的参数就是读进来的x,y，而不是x和y的dfs新编号（这个其他读者应该不会写错，只是我当时脑子一抽写错了。。。）
下面粘一下代码，我的代码其实跑得很慢的，由于我不会卡常，慢到在bzoj上T了，所以标签就没挂bzoj

update:之前用cin读入太慢了，换成scanf就过了，然后加上BZOJ的标签了。

#include <bits/stdc++.h>
using namespace std;
 
int n,b[30001],q,head[60001],cnt;
int tot[60002],son[60001],dep[60002],fa[60002];
int top[60002],ys[60002],yss[60002],z;
struct node
{
	int next,to;
}a[100001];
struct tree
{
	int tag,sum,maxn,r,l;
}tr[1000001];
void add(int from,int to)
{
	a[++cnt].next=head[from];
	a[cnt].to=to;
	head[from]=cnt;	
}
void dfs1(int x)
{
	tot[x]=1;
	for(int i=head[x];i;i=a[i].next)
	{
		int y=a[i].to;
		if(y!=fa[x])
		{
			dep[y]=dep[x]+1;
			fa[y]=x;
			dfs1(y);
			if(tot[y]>tot[son[x]])
			son[x]=y;
			tot[x]+=tot[y];
		}
	}
}
void dfs2(int x,int tp)
{
	top[x]=tp;
	ys[x]=++z;
	yss[z]=x;
	if(son[x]!=0)
	{
		dfs2(son[x],tp);
		for(int i=head[x];i;i=a[i].next)
		{
			int y=a[i].to;
			if(y!=fa[x]&&y!=son[x])
			dfs2(y,y);
		}
	}
}
void update(int num)
{
	tr[num].sum=tr[num<<1].sum+tr[num<<1|1].sum;
	tr[num].maxn=max(tr[num<<1].maxn,tr[num<<1|1].maxn);
}
void build(int num,int l,int r)
{
	tr[num].l=l;
	tr[num].r=r;
	if(l==r)
	{
		tr[num].sum=b[yss[l]];
		tr[num].maxn=b[yss[l]];
		return;
	}
	int mid=(l+r)>>1;
	build(num<<1,l,mid);
	build(num<<1|1,mid+1,r);
	update(num);
}
void change(int num,int p,int c)
{
	if(tr[num].l>p||tr[num].r<p)
	return;
	if(tr[num].l==tr[num].r)
	{
		tr[num].sum=c;
		tr[num].maxn=c;
		return;
	}
	change(num<<1,p,c);
	change(num<<1|1,p,c);
	update(num);
}
int query(int num,int x,int y)
{
	if(y<tr[num].l||x>tr[num].r)
	return 0;
	if(tr[num].l>=x&&tr[num].r<=y)
	return tr[num].sum;
	return query(num<<1,x,y)+query(num<<1|1,x,y);
}
int query2(int num,int x,int y)
{
	if(y<tr[num].l||x>tr[num].r)
	return -2000000000;
	if(tr[num].l>=x&&tr[num].r<=y)
	return tr[num].maxn;
	return max(query2(num<<1,x,y),query2(num<<1|1,x,y));
}
int fnd(int x,int y)
{
	int f1=top[x],f2=top[y];
	int ans=0;
	while(f1!=f2)
	{
		if(dep[f1]<dep[f2])
		{
			swap(f1,f2);
			swap(x,y);
		}
		ans+=query(1,ys[f1],ys[x]);
		x=fa[f1];
		f1=top[x];
	}
	if(dep[x]<dep[y])
	ans+=query(1,ys[x],ys[y]);
	else
	ans+=query(1,ys[y],ys[x]);
	return ans;
}
int find(int x,int y)
{
	int f1=top[x],f2=top[y];
	int ans=-2000000000;
	while(f1!=f2)
	{
		if(dep[f1]<dep[f2])
		{
			swap(f1,f2);
			swap(x,y);
		}
		ans=max(ans,query2(1,ys[f1],ys[x]));
		x=fa[f1];
		f1=top[x];
	}
	if(dep[x]<dep[y])
	{
		ans=max(ans,query2(1,ys[x],ys[y]));
	}	
	else
	{
		ans=max(ans,query2(1,ys[y],ys[x]));
	}	
	return ans;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n-1;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	for(int i=1;i<=n;i++)
	scanf("%d",&b[i]);
	dfs1(1);
	dfs2(1,1);
	build(1,1,n);
	scanf("%d",&q);
	for(int i=1;i<=q;i++)
	{
		char s[10];
		int x,y;
		scanf("%s",s);
		scanf("%d%d",&x,&y);
		if(s[0]=='C')
		{
			change(1,ys[x],y);
		}
		if(s[1]=='M')
		{
			printf("%d\n",find(x,y));
		}
		if(s[1]=='S')
		{
			printf("%d\n",fnd(x,y));
		}
	}
	return 0;
}