树的统计（树链剖分）

树的统计（树链剖分）

一树上有 n 个节点，编号分别为 1 到 n，每个节点都有一个权值 w。我们将以下面的形式来要求你对这棵树完成一些操作：

CHANGE u t ：把节点 u 权值改为 t；
QMAX u v ：询问点 u 到点 v 路径上的节点的最大权值；
QSUM u v ：询问点 u 到点 v 路径上的节点的权值和。
注意：从点 u 到点 v 路径上的节点包括 u 和 v 本身。

输入格式
第一行为一个数 n，表示节点个数；

接下来 n−1 行，每行两个整数 a,b，表示节点 a 与节点 b 之间有一条边相连；

接下来一行 n 个整数，第 i 个整数 wi表示节点 i 的权值；

接下来一行，为一个整数 q ，表示操作总数；

接下来 q 行，每行一个操作，以 CHANGE u t 或 QMAX u v 或 QSUM u v的形式给出。

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

样例
Input
4
1 2
2 3
4 1
4 2 1 3
12
QMAX 3 4
QMAX 3 3
QMAX 3 2
QMAX 2 3
QSUM 3 4
QSUM 2 1
CHANGE 1 5
QMAX 3 4
CHANGE 3 6
QMAX 3 4
QMAX 2 4
QSUM 3 4

Output
4
1
2
2
10
6
5
6
5
16

数据范围与提示
对于 100% 的数据，有 1≤n≤3×10⁴ ,0≤q≤2×10⁵。中途操作中保证每个节点的权值 w 在 −30000 至 30000 之间。

树链剖分板

AC代码：

#include<stdio.h>
#include<string.h>
#define ll long long
#define lef st,mid,rt<<1
#define rig mid+1,en,rt<<1|1
#include<algorithm>
using namespace std;
//typedef long long ll;
const int N=4e5+10;
char c[10];
ll n,q,tim,cnt,l,r;
ll h[N],w[N],head[N],ne[N],to[N],dfn[N];
ll tr[N],maxx[N],nw[N],fa[N],sz[N],son[N],top[N];
//dfn[i]结点i的dfs序（dfs序就相当于线段树的结点编号）
//nw[i]在dfs序下的当前点的权值
//sz[i]表示以i为根节点的树的节点个数
//son[i]表示i的重儿子
//top[i]表示结点i所在链的顶端
void add(ll x,ll y)
{
	to[cnt]=y;
	ne[cnt]=head[x];
	head[x]=cnt++;
}
void ud(ll rt)
{
	tr[rt]=tr[rt<<1]+tr[rt<<1|1];
	maxx[rt]=max(maxx[rt<<1],maxx[rt<<1|1]);
}
void build(ll st,ll en,ll rt)
{
	if(st==en)
	tr[rt]=nw[en],maxx[rt]=nw[en];
	else
	{
		ll mid=(st+en)>>1;
		build(lef);
		build(rig);
		ud(rt);
	}
}
void update(ll st,ll en,ll rt,ll x,ll val)
{
	if(st==en)
	tr[rt]=val,maxx[rt]=val;
	else
	{
		ll mid=(st+en)>>1;
		if(x<=mid)
		update(lef,x,val);
		else
		update(rig,x,val);
		ud(rt);
	}
}
ll query1(ll st,ll en,ll rt)
{
	if(l<=st&&en<=r)
	return tr[rt];
	ll mid=(st+en)>>1;
	if(r<=mid)
	return query1(lef);
	else if(l>mid)
	return query1(rig);
	else
	{
		ll sum=0;
		sum+=query1(lef);
		sum+=query1(rig);
		return sum;
	}
}
ll query2(ll st,ll en,ll rt)
{
	if(l<=st&&en<=r)
	return maxx[rt];
	ll mid=(st+en)>>1;
	if(r<=mid)
	return query2(lef);
	else if(l>mid)
	return query2(rig);
	else
	{
		ll suml=query2(lef);
		ll sumr=query2(rig);
		return max(suml,sumr);
	}
}
ll  querymax(ll u,ll v)
{
	ll ans=-0x3f3f3f3f;
	while(top[u]!=top[v])//u和v不在一条链上
	{
		if(h[top[u]]<h[top[v]])//顶部高度
		swap(u,v);
		l=dfn[top[u]];//查询当前这条小链（小区间）
		r=dfn[u];
		ll sum=query2(1,n,1);
		u=fa[top[u]];//往上跳，直到uv在同一条链上
		ans=max(ans,sum);
	}
	//现在在一条链上了
	if(h[u]<h[v])//通过判断深度来判断那是左边界那是右边界
	swap(u,v);
	l=dfn[v],r=dfn[u];
	ll sum=query2(1,n,1);
	ans=max(ans,sum);//printf("%lld+++\n",ans);
	return ans;
}
ll querysum(ll u,ll v)//同上
{
	ll ans=0;
	while(top[u]!=top[v])
	{
		if(h[top[u]]<h[top[v]])
		swap(u,v);
		l=dfn[top[u]];
		r=dfn[u];
		ans+=query1(1,n,1);
		u=fa[top[u]];
	}
	if(h[u]<h[v])
	swap(u,v);
	l=dfn[v];
	r=dfn[u];
	ans+=query1(1,n,1);
	return ans;
}
//求这棵树的所有重儿子，所有点的深度，和所有点的父亲
void dfs1(ll x,ll fu)
{
	fa[x]=fu;//x的父亲为fu
	h[x]=h[fu]+1;//x的深度为fu的深度+1
	sz[x]=1;//起初以x为根的树的结点只有他本身
	ll mxsize=-1;//判断是否为重儿子的临时变量
	for(ll i=head[x];~i;i=ne[i])//遍历所有儿子
	{
		if(to[i]==fu)//无向边所以要避免父亲
		continue;
		dfs1(to[i],x);
		sz[x]+=sz[to[i]];//+子节点的size
		if(sz[to[i]]>mxsize)//更新重儿子
		{
			mxsize=sz[to[i]];
			son[x]=to[i];
		}
	}
}
//我们从根开始一遍DFS，如果有重儿子就先走重儿子（将一条重链放到一个区间内），搜到叶子节点后回溯回去再去搜轻儿子（轻链）
void dfs2(ll u,ll t)//u所在链的顶端为t
{
	dfn[u]=++tim;//时间戳，相当于给这个点一个新的编号（dfs序）
	nw[tim]=w[u];//在dfs序下的权值
	top[u]=t;//记录u所在链的顶端为t
	if(!son[u])//没有重儿子，说明是叶子结点（有儿子就一定有重儿子）
	return ;
	dfs2(son[u],t);//u的重儿子和u在同一条链上
	for(ll i=head[u];~i;i=ne[i])//搜轻儿子
	{
		if(to[i]==fa[u]||to[i]==son[u])//是轻儿子
		continue;
		dfs2(to[i],to[i]);//以轻儿子为顶端的链
	}
}
int main()
{
	memset(head,-1,sizeof(head));
	scanf("%lld",&n);
	ll i,j,k,a,b;
	for(i=0;i<n-1;i++)
	{
		scanf("%lld%lld",&a,&b);
		add(a,b),add(b,a);
	}
	for(i=1;i<=n;i++)
	scanf("%lld",&w[i]);
	dfs1(1,0);
	dfs2(1,1);
	build(1,n,1);
	scanf("%lld",&q);
	while(q--)
	{
		ll u,v;	
		scanf("%s %lld%lld",c,&u,&v);
		if(strcmp(c,"QMAX")==0)
		printf("%lld\n",querymax(u,v));
		else if(strcmp(c,"QSUM")==0)
		printf("%lld\n",querysum(u,v));
		else
		update(1,n,1,dfn[u],v);
	}
	
	return 0;
}