P3178 [HAOI2015] 树上操作 树链剖分

给出一个根为1，$n\le 1e5$的树，然后支持单点$+d$，区间$+d$，查询某一个结点$u$到$1$路径上的点权之和。
还是直接树剖，线段树维护单点加值和区间加值就可以了。

#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;
int n,q;
int w[N];
vector<int> G[N];

int sz[N],dep[N],fa[N],son[N];
int id[N],a[N],bel[N],cnt=0;
void dfs1(int u,int f,int d) {
	sz[u]=1,dep[u]=d,fa[u]=f,son[u]=-1;
	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]=w[u],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];
ll 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]+=1LL*lazy[rt]*(mid-l+1);
		sum[rt<<1|1]+=1LL*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);
}
ll ask(int x,int y) {
	ll ans=0;
	while(bel[x]!=bel[y]) {
		if(dep[bel[x]]<dep[bel[y]]) swap(x,y);
		ans+=query(1,1,n,id[bel[x]],id[x]);
		x=fa[bel[x]];
	}
	if(dep[x]>dep[y]) swap(x,y);
	ans+=query(1,1,n,id[x],id[y]);
	return ans; 
}
int main() {
	scanf("%d%d",&n,&q);
	for(int i=1;i<=n;i++)
		scanf("%d",&w[i]);
	for(int i=1;i<n;i++) {
		int u,v;
		scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	dfs1(1,0,1);
	dfs2(1,1);
	build(1,1,n); 
	while(q--) {
		int opt,u,v,d;
		scanf("%d",&opt);
		if(opt==1) {
			scanf("%d%d",&u,&d);
			modify(1,1,n,id[u],id[u],d);
		}
		else if(opt==2) {
			scanf("%d%d",&u,&d);
			modify(1,1,n,id[u],id[u]+sz[u]-1,d);
		}
		else if(opt==3) {
			scanf("%d",&u);
			printf("%lld\n",ask(u,1));
		}
	}
	return 0;
}