bzoj4034 cogs1963 [HAOI2015]树上操作

dfs序+线段树
先求出每个点的出栈入栈时间
然后用线段树去维护
入栈为正出栈为负
那么区间和就可以表示一个子树的和
操作1是简单的单点修改
操作2是简单的区间修改
对于操作3 我们可以发现 从根到一个点的路径在dfs序上是连续的一段 那么我们直接查询1到in[x]就可以了

听说这道题分块也可以做？ 没有考虑 另外就是树剖也可以做这道题

#include <cstdio>
#include <cstring>

using namespace std;

typedef long long LL;

const long long N=400005;

long long n=0,m=0;
long long a[N],b[N];
long long head[N],next[N*2],to[N*2],edge=0;
long long t=0,in[N],out[N];
long long isIn[N];
long long flag[N<<2];
LL sum[N<<2],plus[N<<2];

inline void addEdge(long long u,long long v) {
    to[edge]=v,next[edge]=head[u],head[u]=edge++;
    to[edge]=u,next[edge]=head[v],head[v]=edge++;   
}

void dfs(long long x,long long fa) {
    b[in[x]=++t]=a[x];
    isIn[t]=1;
    for (long long e=head[x];~e;e=next[e]) {
        long long v=to[e];
        if (v!=fa)
            dfs(v,x);
    }
    b[out[x]=++t]=-a[x];
    isIn[t]=0;      
}

inline void pushUp(long long rt) {
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];    
}

inline void pushDown(long long rt) {
    if (plus[rt]) {
        sum[rt<<1]+=flag[rt<<1]*plus[rt];
        sum[rt<<1|1]+=flag[rt<<1|1]*plus[rt];
        plus[rt<<1]+=plus[rt];
        plus[rt<<1|1]+=plus[rt];
        plus[rt]=0;
    }   
}

void build(long long rt,long long l,long long r) {
    if (l==r) {
        flag[rt]=isIn[l]?1:-1;
        sum[rt]=b[l];
    } else {
        long long mid=(l+r)>>1;
        build(rt<<1,l,mid);
        build(rt<<1|1,mid+1,r);
        flag[rt]=flag[rt<<1]+flag[rt<<1|1];
        pushUp(rt);
    }
}

void update(long long rt,long long l,long long r,long long l0,long long r0,long long val) {
    if (l0<=l && r<=r0) {
         plus[rt]+=val;
         sum[rt]+=flag[rt]*val;
    } else {
        pushDown(rt);
        long long mid=(l+r)>>1;
        if (l0<=mid)
            update(rt<<1,l,mid,l0,r0,val);
        if (r0>mid)
            update(rt<<1|1,mid+1,r,l0,r0,val);
        pushUp(rt);
    }
}

LL query(long long rt,long long l,long long r,long long l0,long long r0) {
    if (l0<=l && r<=r0) {
        return sum[rt];
    } else {
        pushDown(rt);
        long long mid=(l+r)>>1;
        LL ret=0;
        if (l0<=mid)
            ret+=query(rt<<1,l,mid,l0,r0);
        if (r0>mid)
            ret+=query(rt<<1|1,mid+1,r,l0,r0);
        return ret;     
    }
}

int main(void) {
    freopen("haoi2015_t2.in","r",stdin);
    freopen("haoi2015_t2.out","w",stdout);
    memset(head,-1,sizeof(head));
    scanf("%lld%lld",&n,&m);
    for (long long i=1;i<=n;++i)
        scanf("%lld",a+i);
    for (long long i=0;i<n-1;++i) {
        long long u=0,v=0;
        scanf("%lld%lld",&u,&v);
        addEdge(u,v);
    }       
    dfs(1,-1);
    build(1,1,n<<1);
    while (m--) {
        long long type=0,arg1=0;
        scanf("%lld%lld",&type,&arg1);
        switch (type) {
            case 1:{
                long long arg2=0;
                scanf("%lld",&arg2);
                update(1,1,n<<1,in[arg1],in[arg1],arg2);
                update(1,1,n<<1,out[arg1],out[arg1],arg2);
                break;
            }
            case 2:{
                long long arg2=0;
                scanf("%lld",&arg2);
                update(1,1,n<<1,in[arg1],out[arg1],arg2);
                break;
            }
            case 3:{
                printf("%lld\n",query(1,1,n<<1,1,in[arg1]));
                break;
            }
        }       
    }
    fclose(stdin);
    fclose(stdout);
    return 0;
}