[BZOJ4034][HAOI2015]T2

原题地址

树剖维护子树…

考虑树链剖分，节点按访问顺序成为一个序列，那么我们发现节点x及其子树是在一段连续的区间内，那么就可以用线段树维护子树了.

所以树剖既能维护链又能维护子树(而LCT只能维护链).

这题写的时候普通线段树手滑调了好久…完蛋了…

AC code:

#include <cstdio>
#include <vector>
using namespace std;
typedef long long ll;
const int N=100010;
int n,m,cnt,tot;
int a[N],b[N],siz[N],pre[N],num[N],top[N];
vector<int> G[N];

struct nod{
    int l,r;
    ll  sum,tag;
    nod *lc,*rc;
}pool[N<<2];

struct Segtree{
    nod *root;

    Segtree(){
        root=&pool[tot++];
        build(&root,1,n);
    }

    void pushdown(nod *p){
        p->sum+=p->tag*(p->r-p->l+1);
        if(p->lc) p->lc->tag+=p->tag,p->rc->tag+=p->tag;
        p->tag=0;
    }
    void build(nod **p,int L,int R){
        *p=&pool[tot++];
        (*p)->l=L;(*p)->r=R;
        if(L==R){
            (*p)->sum=b[L];
            (*p)->lc=(*p)->rc=NULL;
            return ;
        }
        int M=(L+R)>>1;
        build(&(*p)->lc,L,M);
        build(&(*p)->rc,M+1,R);
        (*p)->sum=(*p)->lc->sum+(*p)->rc->sum;
    }
    void add(nod *p,int L,int R,int v){
        pushdown(p);
        if(p->l==L&&p->r==R){
            p->tag+=v;
            return ;
        }
        int M=(p->l+p->r)>>1;
        if(R<=M) add(p->lc,L,R,v);
        else if(L>M) add(p->rc,L,R,v);
        else{
            add(p->lc,L,M,v);
            add(p->rc,M+1,R,v);
        }
        pushdown(p->lc);pushdown(p->rc);
        p->sum=p->lc->sum+p->rc->sum;
    }
    ll getsum(nod *p,int L,int R){
        pushdown(p);
        if(p->l==L&&p->r==R) return p->sum;
        int M=(p->l+p->r)>>1;
        if(R<=M) return getsum(p->lc,L,R);
        else if(L>M) return getsum(p->rc,L,R);
        else return getsum(p->lc,L,M)+getsum(p->rc,M+1,R);
    }
};

void dfs1(int x,int pr){
    siz[x]=1;pre[x]=pr;
    for(int i=0;i<(int)G[x].size();i++){
        int y=G[x][i];
        if(y==pre[x]) continue;
        dfs1(y,x);
        siz[x]+=siz[y];
    }
}

void dfs2(int x,int tp){
    num[x]=++cnt;top[x]=tp;
    int mx=0,y=0;
    for(int i=0;i<(int)G[x].size();i++){
        int z=G[x][i];
        if(z!=pre[x]&&siz[z]>mx){
            y=z;
            mx=siz[z];
        }
    }
    if(y) dfs2(y,tp);
    for(int i=0;i<(int)G[x].size();i++){
        int z=G[x][i];
        if(z!=pre[x]&&z!=y) dfs2(z,z);
    }
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<n;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        G[u].push_back(v);G[v].push_back(u);
    }
    dfs1(1,0);
    dfs2(1,1);
    for(int i=1;i<=n;i++) b[num[i]]=a[i];
    Segtree T;
    for(int i=1;i<=m;i++){
        int t,x,a;
        scanf("%d%d",&t,&x);
        if(t==1){
            scanf("%d",&a);
            T.add(T.root,num[x],num[x],a);
        }
        else if(t==2){
            scanf("%d",&a);
            T.add(T.root,num[x],num[x]+siz[x]-1,a);
        }
        else{
            ll sum=0;
            for(int i=x;i;i=pre[top[i]]) sum+=T.getsum(T.root,num[top[i]],num[i]);
            printf("%lld\n",sum);
        }
    }

    return 0;
}