【树链剖分】树上的询问

                                            【题目描述】

给你一棵具有N个点（编号为1到N）M条边的树，并给定各个点权的值，然后有3种操作：
I C1 C2 K：把C1与C2的路径上的所有点权值加上K
D C1 C2 K：把C1与C2的路径上的所有点权值减去K
Q C：查询节点编号为C的权值

【输入格式】

输入包含多组测试数据。
对于每组测试数据的第一行包含三个整数N,M,P，分别表示N个结点(1<=N<=50000)，M条边(M=N-1)和P(1≤P≤100000)个操作。
接下来一行为N个整数A1,A2,...AN(0≤Ai≤1000),表示一开始每个结点的值。
接下来的M行，每行两个整数u和v，表示u到v有一条边。
接下来P行，表示相应的操作，其中0<=k<=1000。

【输出格式】

对于每个询问输出一行，为询问结点的值。

---

树链剖分的裸题，不过对区间的操作比较简单，我们可以直接用树状数组的区间修改与单点查询，没必要写线段树，代码上会简短不少

#include<iostream>
#include<cstdio>
#include<queue>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<bitset>
using namespace std;
int n,m,q,a[50005],h[50005],cnt,size[50005],prt[50005],son[50005],deep[50005],seg[50005],top[50005],rev[50005],tree[50005];
struct edge
{
	int to,next;
}w[100005];
void add(int x,int y)
{
	cnt++;
	w[cnt].to=y;
	w[cnt].next=h[x];
	h[x]=cnt;
}
void dfs(int x,int fa)
{
	prt[x]=fa;
	size[x]=1;
	deep[x]=deep[fa]+1;
	for(int i=h[x];i;i=w[i].next)
	{
		int to=w[i].to;
		if(to==fa) continue;
		dfs(to,x);
		size[x]+=size[to];
		if(size[to]>size[son[x]]) son[x]=to;
	}
}
void dfs2(int x)
{
	if(son[x])
	{
		top[son[x]]=top[x];
		seg[son[x]]=++seg[0];
		rev[seg[0]]=son[x];
		dfs2(son[x]);
	}
	for(int i=h[x];i;i=w[i].next)
	{
		int to=w[i].to;
		if(top[to]) continue;
		top[to]=to;
		seg[to]=++seg[0];
		rev[seg[0]]=to;
		dfs2(to);
	}
}
void add2(int x,int k)
{
//	cout<<x<<endl;
	while(x<=n)
	{
		tree[x]+=k;
		x+=(x&-x);
	}
}
int sum(int x)
{
	int ans=0;
	while(x)
	{
		ans+=tree[x];
		x-=(x&-x);
	}
	return ans;
}
void uprange(int x,int y,int z)
{
//	cout<<x<<" "<<y<<" "<<z<<endl;
	while(top[x]!=top[y])
	{
		if(deep[top[x]]<deep[top[y]]) swap(x,y);
		add2(seg[top[x]],z);
		add2(seg[x]+1,-z);
		x=prt[top[x]];
	}
	if(deep[x]>deep[y]) swap(x,y);
	add2(seg[x],z);
	add2(seg[y]+1,-z);
}
int main()
{
	while(~scanf("%d%d%d",&n,&m,&q))
	{
		memset(tree,0,sizeof(tree));
		for(int i=1;i<=n;i++) scanf("%d",&a[i]);
		for(int i=1;i<=m;i++)
		{
			int x,y;
			scanf("%d%d",&x,&y);
			add(x,y);
			add(y,x);
		}
		dfs(1,0);
		seg[0]=seg[1]=rev[1]=top[1]=1;
		dfs2(1);
		for(int i=1;i<=n;i++) add2(seg[i],a[i]),add2(seg[i]+1,-a[i]);
		for(int i=1;i<=q;i++)
		{
			char s[10];
			scanf("%s",s);
			if(s[0]=='Q')
			{
				int x;
				scanf("%d",&x);
				printf("%d\n",sum(seg[x]));
			}
			else if(s[0]=='D')
			{
				int l,r,z;
				scanf("%d%d%d",&l,&r,&z);
				uprange(l,r,-z);
			}
			else
			{
				int l,r,z;
				scanf("%d%d%d",&l,&r,&z);
				uprange(l,r,z);
			}
		}
	}
}