P3833 [SHOI2012] 魔法树 树链剖分

给出$n\le 1e5$的树，然后两个操作，分别是询问某个点所在的子树的和，以及给$u,v$两个点的路径上加上$d$。
基本上是树剖的模板题，按重链的顺序dfs一遍，子树是一段连续的区间，而树上路径可以在$O(log)$时间里面跳重边维护。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const ll INF=LONG_LONG_MAX;
const int N=1e5+7;
vector<int> G[N]; 

int n,m; 
int sz[N],son[N],dep[N],fa[N];
int id[N],a[N],bel[N];
int cnt=0;  

void dfs1(int u,int f,int d) {
	sz[u]=1,son[u]=-1,dep[u]=d,fa[u]=f;
	for(auto &v:G[u]) {
		if(v==f) continue;
		dfs1(v,u,d+1);
		sz[u]+=sz[v];
		if(son[u]==-1||sz[son[u]]<sz[v]) son[u]=v;
	}
} 
void dfs2(int u,int top) {
	id[u]=++cnt,a[cnt]=0,bel[u]=top;
	if(son[u]==-1) return;
	dfs2(son[u],top);
	for(auto &v:G[u]) {
		if(v==fa[u]||v==son[u]) continue;
		dfs2(v,v);
	}	
} 

ll sum[N<<2];
int lazy[N<<2];

void pushup(int rt) {
	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void pushdown(int rt,int l,int mid,int r) {
	if(lazy[rt]) {
		sum[rt<<1]+=lazy[rt]*(mid-l+1);
		sum[rt<<1|1]+=lazy[rt]*(r-mid);
		lazy[rt<<1]+=lazy[rt];
		lazy[rt<<1|1]+=lazy[rt];
		lazy[rt]=0;
	}
}
void build(int rt,int l,int r) {
	if(l==r) { sum[rt]=a[l]; return; }
	int mid=(l+r)>>1;
	build(rt<<1,l,mid);
	build(rt<<1|1,mid+1,r);
	pushup(rt);
}

void modify(int rt,int l,int r,int L,int R,int x)  {
	if(R<l||L>r) return;
	if(L<=l&&r<=R) {
		sum[rt]+=1LL*x*(r-l+1);
		lazy[rt]+=x;
		return; 
	}
	int mid=(l+r)>>1;
	pushdown(rt,l,mid,r);
	modify(rt<<1,l,mid,L,R,x);
	modify(rt<<1|1,mid+1,r,L,R,x);
	pushup(rt);
} 

ll query(int rt,int l,int r,int L,int R) {
	if(R<l||L>r) return 0;
	if(L<=l&&r<=R) return sum[rt];
	int mid=(l+r)>>1;
	pushdown(rt,l,mid,r);
	return query(rt<<1,l,mid,L,R)+query(rt<<1|1,mid+1,r,L,R);
}

void change(int u,int v,int d) {
	while(bel[u]!=bel[v]) {
		if(dep[bel[u]]<dep[bel[v]]) swap(u,v);
		modify(1,1,n,id[bel[u]],id[u],d);
		u=fa[bel[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	modify(1,1,n,id[u],id[v],d);   	
}


int main() {
	scanf("%d",&n);
	for(int i=1;i<n;i++) {
		int u,v;
		scanf("%d%d",&u,&v);
		u++,v++;
		G[u].push_back(v);
		G[v].push_back(u);
	} 
	dfs1(1,0,1);
	dfs2(1,1);
	build(1,1,n);
	scanf("%d",&m);
	while(m--) {
		char opt[3];
		scanf("%s",opt);
		if(opt[0]=='A') {
			int u,v,d;
			scanf("%d%d%d",&u,&v,&d);
			u++,v++;
			change(u,v,d);
		}
		else if(opt[0]=='Q') {
			int u;
			scanf("%d",&u);
			u++;
			printf("%lld\n",query(1,1,n,id[u],id[u]+sz[u]-1));
		}
	}
	return 0;
}