Aragorn's Story（hdu3966）

题意：给一棵树，并给定各个点权的值，然后有3种操作：

I C1 C2 K: 把C1与C2的路径上的所有点权值加上K

D C1 C2 K：把C1与C2的路径上的所有点权值减去K

Q C：查询节点编号为C的权值

/*
  要手动扩栈。
*/
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<iostream>
#include<cstring>
#define N 100010
using namespace std;
int a[N],head[N],fa[N],son[N],dep[N],top[N],pos[N],sum[N*],tag[N*],n,m,p,sz;
struct node{
    int to,pre;
};node e[N*];
void add(int i,int x,int y){
    e[i].to=y;
    e[i].pre=head[x];
    head[x]=i;
}
void dfs1(int x){
    son[x]=;
    for(int i=head[x];i;i=e[i].pre){
        int v=e[i].to;
        if(fa[x]==v) continue;
        fa[v]=x;dep[v]=dep[x]+;
        dfs1(v);
        son[x]+=son[v];
    }
}
void dfs2(int x,int chain){
    ++sz;pos[x]=sz;top[x]=chain;int k=,maxn=;
    for(int i=head[x];i;i=e[i].pre)
        if(fa[x]!=e[i].to&&son[e[i].to]>maxn){
            maxn=son[e[i].to];
            k=e[i].to;
        }
    if(!k)return;
    dfs2(k,chain);
    for(int i=head[x];i;i=e[i].pre)
        if(e[i].to!=fa[x]&&e[i].to!=k)
            dfs2(e[i].to,e[i].to);
}
void pushup(int k){
    sum[k]=sum[k*]+sum[k*+];
}
void pushdown(int k,int l,int r){
    if(!tag[k]) return;
    int mid=l+r>>;
    tag[k*]+=tag[k];
    tag[k*+]+=tag[k];
    sum[k*]+=tag[k]*(l-mid+);
    sum[k*+]+=tag[k]*(r-mid);
    tag[k]=;
}
void change(int l,int r,int k,int x,int y,int v){
    if(l>=x&&r<=y){
        tag[k]+=v;
        sum[k]+=(l-r+)*v;
        return;
    }
    pushdown(k,l,r);
    int mid=l+r>>;
    if(x<=mid) change(l,mid,k*,x,y,v);
    if(y>mid) change(mid+,r,k*+,x,y,v);
    pushup(k);
}
int query(int l,int r,int k,int pos){
    if(l==r)return sum[k];
    pushdown(k,l,r);
    int mid=l+r>>;
    if(pos<=mid) return query(l,mid,k*,pos);
    else return query(mid+,r,k*+,pos);
}
void xiugai(int x,int y,int v){
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        change(,n,,pos[top[x]],pos[x],v);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y])swap(x,y);
    change(,n,,pos[x],pos[y],v);
}
void work(){
    for(int i=;i<=n;i++)scanf("%d",&a[i]);
    for(int i=;i<=m;i++){
        int x,y;scanf("%d%d",&x,&y);
        add(i*-,x,y);add(i*,y,x);
    }
    dfs1();dfs2(,);
    for(int i=;i<=n;i++)change(,n,,pos[i],pos[i],a[i]);
    char opt[];
    for(int i=;i<=p;i++){
        int x,y,v;
        scanf("%s",opt);
        if(opt[]=='I'){
            scanf("%d%d%d",&x,&y,&v);
            xiugai(x,y,v);
        }
        else if(opt[]=='D'){
            scanf("%d%d%d",&x,&y,&v);
            xiugai(x,y,-v);
        }
        else{
            scanf("%d",&x);
            printf("%d\n",query(,n,,pos[x]));
        }
    }
}
int main(){while(scanf("%d%d%d",&n,&m,&p)!=EOF){
        memset(head,,sizeof(head));
        memset(fa,,sizeof(fa));
        memset(son,,sizeof(son));
        memset(dep,,sizeof(dep));
        memset(top,,sizeof(top));
        memset(pos,,sizeof(pos));
        memset(sum,,sizeof(sum));
        memset(tag,,sizeof(tag));
        sz=;
        work();
    }
    return ;
}