【Bzoj4034】树上操作

来源：互联网发布：在线办公软件qq 编辑：程序博客网时间：2024/06/06 09:05

题意

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

0 0