[FROM WOJ]#1935 魔法树

#1935 魔法树

题面
Harry Potter 新学了一种魔法：可以让改变树上的果子个数。满心欢喜的他找到了一个巨大的果树，来试验他的新法术。
这棵果树共有N个节点，其中节点0是根节点，每个节点u的父亲记为fa[u]，保证有fa[u] < u。初始时，这棵果树上的果子都被 Dumbledore 用魔法清除掉了， 所以这个果树的每个节点上都没有果子（即0个果子）。
不幸的是， Harry 的法术学得不到位，只能对树上一段路径的节点上的果子个数统一增加一定的数量。也就是说， Harry 的魔法可以这样描述：
Add u v d
表示将点u和v之间的路径上的所有节点的果子个数都加上d。
接下来，为了方便检验 Harry 的魔法是否成功，你需要告诉他在释放魔法的过程中的一些有关果树的信息：
Query u
表示当前果树中，以点u为根的子树中，总共有多少个果子？

输入
第一行一个正整数N (1 ≤N≤ 100000)，表示果树的节点总数， 节点以 0, 1, … , N − 1标号， 0一定代表根节点。
接下来N − 1行，每行两个整数a, b (0 ≤ a < b < N)， 表示A是B的父亲。
接下来是一个正整数Q(1 ≤ Q ≤ 100000)，表示共有Q次操作。
后面跟着Q行，每行是以下两种中的一种：

1. A u v d，表示将u到v的路径上的所有节点的果子数加上d；0≤u,v<N,0<d<100000
2. Q u，表示询问以u为根的子树中的总果子数，注意是包括u本身的。 0 ≤ u < N

输出
对于所有的Query操作，依次输出询问的答案，每行一个。 答案可能会超过$2^{32}$，但不会超过$10^{15}$。

样例输入
4
0 1
1 2
2 3
4
A 1 3 1
Q 0
Q 1
Q 2

样例输出
3
3
2

SOL
树链剖分的板子题。
一共两种操作：询问子树和，修改路径信息。
维护就好了，也没什么难度。
注意开long long，还有节点是从0开始的。

代码：

#include<bits/stdc++.h>
#define N 100005
#define int long long
using namespace std;
inline int rd(){
	int data=0;static char ch=0;
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))data=(data<<1)+(data<<3)+ch-'0',ch=getchar();
	return data;
}
inline void write(int x){if(x>9)write(x/10);putchar(x%10+'0');}
int n,m;
char ins;
int cnt,first[N];
struct node{int v,nxt;}e[N<<1];
inline void add(int u,int v){e[++cnt].v=v;e[cnt].nxt=first[u];first[u]=cnt;}
int seg[N<<1],siz[N],son[N],f[N],dep[N],top[N];
void dfs1(int u,int fa,int d){
	siz[u]=1;f[u]=fa;dep[u]=d;
	for(int register i=first[u];i;i=e[i].nxt){
		int register v=e[i].v;
		if(v==fa)continue;
		dfs1(v,u,d+1);
		siz[u]+=siz[v];if(siz[son[u]]<siz[v])son[u]=v;
	}
}
void dfs2(int u,int tp){
	seg[u]=++seg[0];
	top[u]=tp;
	if(!son[u])return;	
	dfs2(son[u],tp);
	for(int register i=first[u];i;i=e[i].nxt){
		int register v=e[i].v;
		if(v==f[u]||v==son[u])continue;
		dfs2(v,v);
	}
}
#define lc (p<<1)
#define rc (p<<1|1)
struct segment_tree{int l,r,sum,add;}t[N<<2];
inline void pushup(int p){t[p].sum=t[lc].sum+t[rc].sum;}
inline void pushdown(int p){
	int register l=t[p].r-t[p].l+1;
	if(!t[p].add)return;
	t[lc].add+=t[p].add;t[rc].add+=t[p].add;
	t[lc].sum+=t[p].add*(l-(l>>1));t[rc].sum+=t[p].add*(l>>1);
	t[p].add=0;
}
inline void build(int p,int l,int r){
	t[p].l=l;t[p].r=r;
	if(l==r)return;
	int register mid=l+r>>1;
	build(lc,l,mid);
	build(rc,mid+1,r);
	pushup(p);
}
inline void change(int p,int v,int ql,int qr){
	int register l=t[p].l,r=t[p].r;
	if(ql<=l&&qr>=r){t[p].sum+=v*(r-l+1);t[p].add+=v;return;}
	int register mid=l+r>>1;
	pushdown(p);
	if(ql<=mid)change(lc,v,ql,qr);
	if(qr>mid)change(rc,v,ql,qr);
	pushup(p);
}
inline int query(int p,int ql,int qr){
	int register l=t[p].l,r=t[p].r,ans=0;
	if(ql<=l&&qr>=r)return t[p].sum;
	int mid=l+r>>1;
	pushdown(p);
	if(ql<=mid)ans+=query(lc,ql,qr);
	if(qr>mid)ans+=query(rc,ql,qr);
	pushup(p);
	return ans;
}
inline void update(int x,int y,int v){
	while(top[x]!=top[y]){
		if(dep[top[x]]<dep[top[y]])swap(x,y);
		change(1,v,seg[top[x]],seg[x]);
		x=f[top[x]];
	}
	if(dep[x]>dep[y])swap(x,y);
	change(1,v,seg[x],seg[y]);
}
signed main(){
	n=rd();
	for(int register i=1;i<n;i++){int register fr=rd(),to=rd();add(fr+1,to+1);add(to+1,fr+1);}
	dfs1(1,0,1);dfs2(1,1);build(1,1,n);
	m=rd();
	while(m--){scanf("\n%c",&ins);
		if(ins=='A'){int register x=rd(),y=rd(),v=rd();++x;++y;update(x,y,v);}
		else{int register x=rd();++x;write(query(1,seg[x],seg[x]+siz[x]-1));puts("");} 
	}
	return 0;
}