题意
有一棵点数为 N 的树,以点 1 为根,且树点有边权。然后有 M 个
操作,分为三种:
操作 1 :把某个节点 x 的点权增加 a 。
操作 2 :把某个节点 x 为根的子树中所有点的点权都增加 a 。
操作 3 :询问某个节点 x 到根的路径中所有点的点权和。
-------------------------------------------------------------------------------------------------------------------------------------------------
树剖的模板题,注意Dfs的时候可以记下每个点的end值。
#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long LL;
const int maxx = 100000 + 25;
int head[maxx],to[maxx<<1],nxt[maxx<<1],end[maxx];
int ftr[maxx],son[maxx],size[maxx],rnk[maxx],dpt[maxx],top[maxx];
LL T[maxx<<2],a[maxx],Add[maxx<<2];
int cnt,num,n,m,x,y,flag;
LL k;
void ins(int x,int y){to[++num]=y;nxt[num]=head[x];head[x]=num;}
namespace Y{
void Dfs1(int x){
size[x] = 1;
for(int i=head[x];i;i=nxt[i]){
int now = to[i];if(now == ftr[x]) continue;
ftr[now] = x;dpt[now] = dpt[x] + 1;
Dfs1(now);size[x] += size[now];
if(size[now] > size[son[x]]) son[x] = now;
}
}
void Dfs2(int x,int brn){
rnk[x] = ++cnt;top[x] = brn;
if(son[x]) Dfs2(son[x],brn);
for(int i=head[x];i;i=nxt[i])
if(to[i] != ftr[x] && to[i] != son[x])
Dfs2(to[i],to[i]);
end[x] = cnt;
}
void pushdown(int i,int l,int r){
int mid = (l+r) >> 1;
T[i<<1]+=Add[i]*(mid-l+1);T[i<<1|1]+=Add[i]*(r-mid);
Add[i<<1]+=Add[i];Add[i<<1|1]+=Add[i];Add[i]=0;
}
void modify(int i,int x,int y,int l,int r,LL k){
if(x<=l && r<=y) {Add[i]+=k;T[i]+=(r-l+1)*k;return;}
int mid = (l+r) >> 1;if(Add[i]) pushdown(i,l,r);
if(x <= mid) modify(i<<1,x,y,l,mid,k);
if(y > mid) modify(i<<1|1,x,y,mid+1,r,k);
T[i] = T[i<<1] + T[i<<1|1];
}
LL Query(int i,int x,int y,int l,int r){
if(x<=l && r<=y) return T[i];LL ans = 0;
int mid = (l+r) >> 1;if(Add[i]) pushdown(i,l,r);
if(x <= mid) ans += Query(i<<1,x,y,l,mid);
if(y > mid) ans += Query(i<<1|1,x,y,mid+1,r);
return ans;
}
LL solve(int x){
LL ans = 0;
while(top[x] != 1)
ans+=Query(1,rnk[top[x]],rnk[x],1,n),x=ftr[top[x]];
ans += Query(1,1,rnk[x],1,n);
return ans;
}
}
using namespace Y;
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
for(int i=1;i<n;i++) scanf("%d%d",&x,&y),ins(x,y),ins(y,x);
Dfs1(1);Dfs2(1,1);
for(int i=1;i<=n;i++) modify(1,rnk[i],rnk[i],1,n,a[i]);
while( m-- ){
scanf("%d",&flag);
if(flag == 1)
scanf("%d%lld",&x,&k),modify(1,rnk[x],rnk[x],1,n,k);
if(flag == 2)
scanf("%d%lld",&x,&k),modify(1,rnk[x],end[x],1,n,k);
if(flag == 3)
scanf("%d",&x),printf("%lld\n",solve(x));
}
return 0;
}