bzoj 4034: [HAOI2015]树上操作(线段树+dfs序)

操作 1 ：把某个节点 x 的点权增加 a 。
操作 2 ：把某个节点 x 为根的子树中所有点的点权都增加 a 。
操作 3 ：询问某个节点 x 到根的路径中所有点的点权和。
用dfs序进栈+1出栈-1+线段树区间修改,查询维护
PS:线段树又写挂了,调试了很久啊~~

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=2e5+10,M=N*4;
typedef long long ll;
ll ls[M],rs[M],flag[M],sum[M],tag[M],a[N],
n,m,
ins[N],out[N],dfssum[N],cnt=0,mk[N];
struct Edge{
    int to,next;
    Edge(int to=0,int next=0):to(to),next(next){}
}e[N<<1];int head[N],tot=0;
void add_edge(int u,int v){e[++tot]=Edge(v,head[u]);head[u]=tot;}
void dfs(int u,int fa)
{
    ins[u]=++cnt;dfssum[cnt]=a[u];mk[cnt]=1;
    for(int i=head[u];i;i=e[i].next){
        int id=e[i].to;
        if(id==fa)continue;
        dfs(id,u);
    }
    out[u]=++cnt;dfssum[cnt]=-a[u];mk[cnt]=-1;
}
void build(int k,int l,int r)
{
    ls[k]=l;rs[k]=r;sum[k]=tag[k]=flag[k]=0;
    if(l==r) {sum[k]=dfssum[l];flag[k]=mk[l];return;}
    int mid=(l+r)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    sum[k]=sum[k<<1]+sum[k<<1|1];
    flag[k]=flag[k<<1]+flag[k<<1|1];
}
void pushdown(int k)
{
    tag[k<<1]+=tag[k];
    tag[k<<1|1]+=tag[k];
    sum[k<<1]+=(tag[k]*flag[k<<1]);
    sum[k<<1|1]+=(tag[k]*flag[k<<1|1]);
    tag[k]=0;
}
void update(int k,int ql,int qr,int delta)
{
    if(ql==ls[k]&&qr==rs[k]){tag[k]+=delta;sum[k]+=(delta*flag[k]);return;}
    if(tag[k])pushdown(k);
    int mid=(ls[k]+rs[k])>>1;
    if(qr<=mid)update(k<<1,ql,qr,delta);
    else if(ql>mid)update(k<<1|1,ql,qr,delta);
    else{
        update(k<<1,ql,mid,delta);
        update(k<<1|1,mid+1,qr,delta);
    }
    sum[k]=sum[k<<1]+sum[k<<1|1];
}
ll query(int k,int ql,int qr)
{
    if(ql==ls[k]&&qr==rs[k]){return sum[k];}
    if(tag[k])pushdown(k);
    int mid=(ls[k]+rs[k])>>1;
    if(qr<=mid)return query(k<<1,ql,qr);
    else if(ql>mid)return query(k<<1|1,ql,qr);
    else return (query(k<<1,ql,mid)+query(k<<1|1,mid+1,qr));
}
int main()
{
    scanf("%lld%lld",&n,&m);
    fo(i,1,n) scanf("%lld",&a[i]);
    for(int u,v,i=1;i<n;i++) {
        scanf("%d%d",&u,&v);
        add_edge(u,v);add_edge(v,u);
    }dfs(1,0);
    build(1,1,cnt);
    for(int op,x,val,i=1;i<=m;i++){
        scanf("%d",&op);
        if(op==1){scanf("%d%d",&x,&val);update(1,ins[x],ins[x],val);update(1,out[x],out[x],val);}
        if(op==2){scanf("%d%d",&x,&val);update(1,ins[x],out[x],val);}
        if(op==3){scanf("%d",&x);printf("%lld\n",query(1,1,ins[x]));}
    }
    return 0;
}