BZOJ4034: [HAOI2015]树上操作(DFS序+线段树+树上问题维护贡献值)

题意：传送门
题解：对于1操作而言，当这个点增加一个值时，那么对于它以及它的子树而言路径的贡献都会加上这个值，这个对于刚开始的值也同样是这个道理，对于2操作而言，它以及它的子树都加上一个值，那么对于某个子树内的点贡献增加了$(de{p}_i-de{p}_x+1)\ast a=de{p}_i\ast a+(1-de{p}_x)\ast a$，对于后面这个$(1-de{p}_x)\ast a$可以用操作一中的方法来维护，对于前面这个可以只维护$a$这个值，最后答案用$de{p}_i$来进行相乘并加以维护。
附上DFS序代码：

void dfs(int u,int f)
{
    in[u]=++cnt;
    for(int i=head[u];~i;i=edges[i].next){
        int v=edges[i].v;
        if(v==f)continue;
        dep[v]=dep[u]+1;
        dfs(v,u);
    }
    out[u]=cnt;
}

附上代码：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=1e5+5;
int n,m,a[maxn],u,v,opt,id,x,cnt,in[maxn],out[maxn],dep[maxn];
struct edge{int u,v,next;}edges[maxn<<1];
int head[maxn],tot;
void init(){memset(head,-1,sizeof(head));tot=0;}
void add_edges(int u,int v)
{
    edges[tot].u=u;edges[tot].v=v;edges[tot].next=head[u];head[u]=tot++;
    edges[tot].u=v;edges[tot].v=u;edges[tot].next=head[v];head[v]=tot++;
}
void dfs(int u,int f)
{
    in[u]=++cnt;
    for(int i=head[u];~i;i=edges[i].next){
        int v=edges[i].v;
        if(v==f)continue;
        dep[v]=dep[u]+1;
        dfs(v,u);
    }
    out[u]=cnt;
}
struct data{
    int l,r,add,mul;
}tree[maxn<<2];
void build(int k,int s,int t)
{
    tree[k].l=s;tree[k].r=t;
    if(s==t)return ;
    int mid=(s+t)>>1;
    build(k<<1,s,mid);
    build(k<<1|1,mid+1,t);
}
void push_down(int k)
{
    tree[k<<1].add+=tree[k].add;
    tree[k<<1|1].add+=tree[k].add;
    tree[k<<1].mul+=tree[k].mul;
    tree[k<<1|1].mul+=tree[k].mul;
    tree[k].add=tree[k].mul=0;
}
void update(int k,int s,int t,int x,int y)
{
    int l=tree[k].l,r=tree[k].r;
    if(l==s&&r==t){
        tree[k].add+=x;tree[k].mul+=y;
        return ;
    }
    int mid=(l+r)>>1;
    push_down(k);
    if(t<=mid)update(k<<1,s,t,x,y);
    else if(s>mid)update(k<<1|1,s,t,x,y);
    else{
        update(k<<1,s,mid,x,y);
        update(k<<1|1,mid+1,t,x,y);
    }
}
int query(int k,int x,int s,int t)
{
    int l=tree[k].l,r=tree[k].r;
    if(l==s&&r==t){
        return dep[x]*tree[k].mul+tree[k].add;
    }
    push_down(k);
    int mid=(l+r)>>1;
    if(t<=mid)return query(k<<1,x,s,t);
    else if(s>mid)return query(k<<1|1,x,s,t);
    else return query(k<<1,x,s,mid)+query(k<<1,x,mid+1,t);
}
int32_t main()
{
    scanf("%lld%lld",&n,&m);
    init();
    for(int i=1;i<=n;i++){
        scanf("%lld",&a[i]);
    }
    for(int i=1;i<n;i++){
        scanf("%lld%lld",&u,&v);
        add_edges(u,v);
    }
    dep[1]=1;
    dfs(1,-1);
    build(1,1,n);
    for(int i=1;i<=n;i++){
        update(1,in[i],out[i],a[i],0);
    }
    for(int i=1;i<=m;i++){
        scanf("%lld",&opt);
        if(opt==1){
            scanf("%lld%lld",&id,&x);
            update(1,in[id],out[id],x,0);
        }else if(opt==2){
            scanf("%lld%lld",&id,&x);
            update(1,in[id],out[id],(1-dep[id])*x,x);
        }else{
            scanf("%lld",&x);
            printf("%lld\n",query(1,x,in[x],in[x]));
        }
    }
    return 0;
}