BZOJ 4034 HAOI2015 T2 DFS序+线段树

题目大意：给定一棵树，每个点有点权，支持下列操作：
1.某个点的点权+a
2.某棵子树所有点权+a
3.查询某个点到根路径上的点权和
这个用入栈出栈序就可以了
入栈为正，出栈为负，那么一个点到根路径上的权值和就是入栈出栈序中[1,入栈位置]的和
而子树在入栈出栈序中是连续的，因此用线段树维护一下就可以了
(似乎只要无脑链剖就可以了？

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 100100
using namespace std;
struct abcd{
    int to,next;
}table[M<<1];
int head[M],tot;
int n,m;
int a[M],f[M<<1];
bool sign[M<<1];
int st[M],ed[M];
void Add(int x,int y)
{
    table[++tot].to=y;
    table[tot].next=head[x];
    head[x]=tot;
}
void DFS(int x,int from)
{
    static int T;
    int i;
    f[st[x]=++T]=a[x];
    sign[T]=false;
    for(i=head[x];i;i=table[i].next)
        if(table[i].to!=from)
            DFS(table[i].to,x);
    f[ed[x]=++T]=-a[x];
    sign[T]=true;
}
struct Segtree{
    Segtree *ls,*rs;
    long long sum,mark;
    int flag;
    void* operator new (size_t)
    {
        static Segtree mempool[M<<2],*C=mempool;
        return C++;
    }
    void Add(long long x)
    {
        sum+=flag*x;
        mark+=x;
    }
    void Push_Up()
    {
        sum=ls->sum+rs->sum;
    }
    void Push_Down()
    {
        if(!mark) return ;
        ls->Add(mark);
        rs->Add(mark);
        mark=0;
    }
    void Build_Tree(int x,int y)
    {
        int mid=x+y>>1;
        if(x==y)
        {
            sum=f[mid];
            flag=sign[mid]?-1:1;
            return ;
        }
        (ls=new Segtree)->Build_Tree(x,mid);
        (rs=new Segtree)->Build_Tree(mid+1,y);
        flag=ls->flag+rs->flag;
        Push_Up();
    }
    void Add(int x,int y,int l,int r,long long val)
    {
        int mid=x+y>>1;
        if(x==l&&y==r)
        {
            Add(val);
            return ;
        }
        Push_Down();
        if(r<=mid)
            ls->Add(x,mid,l,r,val);
        else if(l>mid)
            rs->Add(mid+1,y,l,r,val);
        else
            ls->Add(x,mid,l,mid,val) , rs->Add(mid+1,y,mid+1,r,val);
        Push_Up();
    }
    long long Query(int x,int y,int l,int r)
    {
        int mid=x+y>>1;
        if(x==l&&y==r)
            return sum;
        Push_Down();
        if(r<=mid)
            return ls->Query(x,mid,l,r);
        if(l>mid)
            return rs->Query(mid+1,y,l,r);
        return ls->Query(x,mid,l,mid) + rs->Query(mid+1,y,mid+1,r);
    }
}*tree=new Segtree;
int main()
{
    int i,p,x,y;
    cin>>n>>m;
    for(i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        Add(x,y);Add(y,x);
    }
    DFS(1,0);
    tree->Build_Tree(1,n<<1);
    for(i=1;i<=m;i++)
    {
        scanf("%d",&p);
        if(p==1)
        {
            scanf("%d%d",&x,&y);
            tree->Add(1,n<<1,st[x],st[x],y);
            tree->Add(1,n<<1,ed[x],ed[x],y);
        }
        else if(p==2)
        {
            scanf("%d%d",&x,&y);
            tree->Add(1,n<<1,st[x],ed[x],y);
        }
        else
        {
            scanf("%d",&x);
            printf("%lld\n",tree->Query(1,n<<1,1,st[x]));
        }
    }
    return 0;
}