【树链剖分】树上的询问

                                            【题目描述】

给你一棵具有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
#include
#include
#include
#include
#include
#include
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<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);
			}
		}
	}
}