HDU 3966 基础树链剖分

题意：给一棵树，并给定各个点权的值，然后有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 <iostream>
 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 using namespace std;
 #define N 50010
 #define ls o<<1
 #define rs o<<1|1
 #define define_m int m=(l+r)>>1
 #define ll long long
 int n , m , q , first[N] , k;

 struct Edge{
     int y , next;
     Edge(){}
     Edge(int y , int next):y(y),next(next){}
 }e[N<<];

 void add_edge(int x , int y)
 {
     e[k] = Edge(y , first[x]);
     first[x] = k++;
 }

 int sz[N] , fa[N] , son[N] , top[N] , dep[N] , id[N] , number;
 void dfs(int u , int f , int d)
 {
     fa[u] = f;
     sz[u] =  , dep[u] = d , son[u] =;
     int maxn = ;
     for(int i=first[u] ; ~i ; i=e[i].next){
         int v= e[i].y;
         if(v == f) continue;
         dfs(v , u , d+);
         sz[u] += sz[v];
         if(sz[v]>maxn) son[u] = v , maxn=sz[v];
     }
 }

 void dfs1(int u , int f , int head)
 {
     top[u] = head;
     if(son[u]){
         id[son[u]] = ++number;
         dfs1(son[u] , u , head);
     }
     for(int i=first[u] ; ~i ; i=e[i].next){
         int v = e[i].y;
         if(v == f || v == son[u]) continue;
         id[v] = ++number;
         dfs1(v , u , v);
     }
 }

 int val[N] , add[N<<] , siz[N<<];
 ll sum[N<<];

 void push_up(int o){sum[o] = sum[ls]+sum[rs];}

 void push_down(int o)
 {
     if(add[o]){
         add[ls] += add[o] , add[rs] += add[o];
         sum[ls] += (ll)add[o]*siz[ls] , sum[rs] += (ll)add[o]*siz[rs];
         add[o] = ;
     }
 }

 void build(int o , int l , int r)
 {
     add[o] =  , siz[o] = r-l+;
     if(l==r){
         sum[o] = (ll)val[l];
         return ;
     }
     define_m;
     build(ls , l , m);
     build(rs , m+ , r);
     push_up(o);
 }

 void update(int o , int l , int r , int s , int t , int v)
 {
     if(l>=s && r<=t){
         sum[o] += (ll)siz[o]*v;
         add[o] += v;
         return ;
     }
     define_m;
     push_down(o);
     if(m>=s) update(ls , l , m , s , t , v);
     if(m<t) update(rs , m+ , r , s , t , v);
     push_up(o);
 }

 ll query(int o , int l , int r , int p)
 {
     if(l==r && l==p) return sum[o];
     push_down(o);
     define_m;
     if(m>=p) return query(ls , l , m, p);
     else return query(rs , m+ , r , p);
 }

 void updatePath(int u , int v , int change)
 {
     int top1 = top[u] , top2 = top[v];
     while(top1!=top2){
         if(dep[top1]<dep[top2]){
             swap(top1 , top2);
             swap(u , v);
         }
         update( ,  , n , id[top1] , id[u] , change);
         u = fa[top1];
         top1 = top[u];
     }

     if(dep[u]<dep[v]) swap(u , v);
     update( ,  , n , id[v] , id[u] , change);

 }

 char str[];

 int main()
 {
    // freopen("in.txt" , "r" , stdin);
     while(~scanf("%d%d%d" ,&n , &m , &q))
     {
         for(int i= ; i<=n ; i++) scanf("%d" , val+i);
         memset(first , - , sizeof(first));
         k = ;
         while(m--){
             int u , v;
             scanf("%d%d" , &u , &v);
             add_edge(u,v);
             add_edge(v,u);
         }
         number = ;
         dfs( ,  , );
         id[] = ++number;
         dfs1( ,  , );
       //  for(int i=1 ; i<=n ; i++) cout<<i<<" "<<son[i]<<" "<<id[i]<<endl;
         int tmp[N];
         for(int i= ; i<=n ; i++) tmp[i]=val[i];
         for(int i= ; i<=n ; i++) val[id[i]] = tmp[i];
         build( ,  , n);
         for(int i= ; i<q ; i++){
             int s , t , v;
             scanf("%s" , str);
             if(str[] == 'Q'){
                 scanf("%d" , &v);
                 ll ans = query( ,  , n , id[v]);
                 printf("%d\n" , (int)ans);
             }else if(str[]=='I'){
                 scanf("%d%d%d" , &s , &t , &v);
                 updatePath(s , t , v);
             }else{
                 scanf("%d%d%d" , &s , &t , &v);
                 updatePath(s , t , -v);
             }
         }
     }
     return ;
 }