hdu 3966 Aragorn's Story 树链剖分

树链剖分板题。

#pragma comment(linker, "/STACK:102400000,102400000")
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<vector>
#define maxn 50010
using namespace std;
int dep[maxn],son[maxn],fa[maxn],top[maxn],id[maxn],num,size[maxn],a[maxn];
vector<int >g[maxn];
struct pi{
    int le,ri;
    int lazy;
    int sum;
}pp[4*maxn];
void build(int tot,int l,int r){
    pp[tot].sum=0;
    pp[tot].le=l;
    pp[tot].ri=r;
    pp[tot].lazy=0;
    if(l==r) return ;
    build(2*tot,l,(l+r)/2);
    build(2*tot+1,(l+r)/2+1,r);
}
void merg(int tot,int l,int r,int k){
    if(pp[tot].le>=l&&pp[tot].ri<=r){
        pp[tot].lazy+=k;
        return ;
    }
    int mid;
    mid=(pp[tot].le+pp[tot].ri)/2;
    if(l<=mid) merg(2*tot,l,r,k);
    if(r>mid) merg(2*tot+1,l,r,k);
    pp[tot].sum+=(min(pp[tot].ri,r)-max(pp[tot].le,l)+1)*k;
}
int query(int tot,int l,int r){
    if(pp[tot].le>=l&&pp[tot].ri<=r) return pp[tot].sum+pp[tot].lazy*(pp[tot].ri-pp[tot].le+1);
    int mid;
    mid=(pp[tot].le+pp[tot].ri)/2;
    int s=0;
    s=pp[tot].lazy*(min(pp[tot].ri,r)-max(pp[tot].le,l)+1);
    if(l<=mid) s+=query(2*tot,l,r);
    if(r>mid) s+=query(2*tot+1,l,r);
    return s;
}
void dfs1(int u,int pa,int d){
    dep[u]=d;
    size[u]=1;
    fa[u]=pa;
    son[u]=0;
    int i,k,v;
    k=(int)g[u].size();
    for(i=0;i<k;i++){
        v=g[u][i];
        if(v==pa) continue;
        dfs1(v,u,d+1);
        size[u]+=size[v];
        if(size[son[u]]<size[v]){
            son[u]=v;
        }
    }
}
void dfs2(int u,int pa,int tp){
    id[u]=++num;
    top[u]=tp;
    int i,k,v;
    k=(int)g[u].size();
    if(son[u]) dfs2(son[u],u,tp);
    for(i=0;i<k;i++){
        v=g[u][i];
        if(v==pa||v==son[u]) continue;
        dfs2(v,u,v);
    }
}
void update(int u,int v,int k){
    int to1,to2;
    to1=top[u];
    to2=top[v];
    while(to1!=to2){
        if(dep[to1]<dep[to2]){
            swap(u,v);
            swap(to1,to2);
        }
        merg(1,id[to1],id[u],k);
        u=fa[to1];
        to1=top[u];
    }
    if(dep[u]>dep[v]) swap(u,v);
    merg(1,id[u],id[v],k);
}
char c[5];
int main()
{
    int n,m,p,i,j,k,f;
    while(scanf("%d%d%d",&n,&m,&p)!=EOF){
        build(1,1,n);
        for(i=1;i<=n;i++) g[i].clear();
        for(i=1;i<=n;i++){
            scanf("%d",&a[i]);
        }
        for(i=0;i<m;i++){
            scanf("%d%d",&k,&f);
            g[k].push_back(f);
            g[f].push_back(k);
        }
        num=0;
        dfs1(1,1,1);
        dfs2(1,1,1);
        for(i=1;i<=n;i++){
            merg(1,id[i],id[i],a[i]);
        }
        for(i=0;i<p;i++){
            scanf("%s",c);
            if(c[0]=='I'){
                scanf("%d%d%d",&k,&f,&j);
                update(k, f, j);
            }
            else if(c[0]=='D'){
                scanf("%d%d%d",&k,&f,&j);
                update(k, f, -j);
            }
            else{
                scanf("%d",&k);
                printf("%d\n",query(1,id[k],id[k]));
            }
        }
    }
}