【Bzoj4034】树上操作

题意

有一棵点数为 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;
}