BZOJ 1036 [ZJOI2008]树的统计Count (树链剖分 - 点权剖分 - 单点权修改)

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1036

树链剖分模版题，打的时候注意点就行。做这题的时候，真的傻了，单词拼错检查了一个多小时...

代码如下：

 //树链剖分 点权修改 修改单节点
 #include <iostream>
 #include <cstring>
 #include <algorithm>
 #include <cstdio>
 using namespace std;
 const int MAXN = 3e4 + ;
 struct data {
     int to , next;
 }edge[MAXN << ];
 int head[MAXN] , cnt , tot;
 int top[MAXN] , par[MAXN] , son[MAXN] , size[MAXN] , dep[MAXN];
 int id[MAXN] , fid[MAXN]; //id[i]表示i对应在线段树上的位置 fid[i]表示线段树位置是i的叶子 对应的节点
 int a[MAXN];

 void init() {
     tot = cnt = ;
     memset(head , - , sizeof(head));
 }

 inline void add(int u , int v) {
     edge[tot].next = head[u];
     edge[tot].to = v;
     head[u] = tot++;
 }

 void dfs1(int u , int p , int d) {
     dep[u] = d , size[u] =  , son[u] = u , par[u] = p;
     for(int i = head[u] ; ~i ; i = edge[i].next) {
         int v = edge[i].to;
         if(v == p)
             continue;
         dfs1(v , u , d + );
         if(size[v] >= size[son[u]] || son[u] == u)
             son[u] = v;
         size[u] += size[v];
     }
 }

 void dfs2(int u , int p , int t) {
     top[u] = t , id[u] = ++cnt;
     fid[cnt] = u;
     if(son[u] != u)
         dfs2(son[u] , u , t);
     for(int i = head[u] ; ~i ; i = edge[i].next) {
         int v = edge[i].to;
         if(v == p || v == son[u])
             continue;
         dfs2(v , u , v);
     }
 }

 struct segtree {
     int l , r;
     int sum , Max;
 }T[MAXN << ];

 void build(int p , int l , int r) {
     int mid = (l + r) >> ;
     T[p].l = l , T[p].r = r;
     if(l == r) {
         T[p].Max = T[p].sum = a[fid[l]]; //
         return ;
     }
     build(p <<  , l , mid);
     build((p << )| , mid +  , r);
     T[p].sum = T[p << ].sum + T[(p << )|].sum;
     T[p].Max = max(T[p << ].Max , T[(p << )|].Max);
 }

 void updata(int p , int pos , int num) {
     int mid = (T[p].l + T[p].r) >> ;
     if(T[p].l == T[p].r && T[p].l == pos) {
         T[p].sum = T[p].Max = num;
         return ;
     }
     if(pos <= mid) {
         updata(p <<  , pos , num);
     }
     else {
         updata((p << )| , pos , num);
     }
     T[p].sum = T[p << ].sum + T[(p << )|].sum;
     T[p].Max = max(T[p << ].Max , T[(p << )|].Max);
 }

 int query_sum(int p , int l , int r) {
     int mid = (T[p].l + T[p].r) >> ;
     if(T[p].l == l && T[p].r == r) {
         return T[p].sum;
     }
     if(r <= mid) {
         return query_sum(p <<  , l , r);
     }
     else if(l > mid) {
         return query_sum((p << )| , l , r);
     }
     else {
         return query_sum(p <<  , l , mid) + query_sum((p << )| , mid +  , r);
     }
 }

 int query_max(int p , int l , int r) {
     int mid = (T[p].l + T[p].r) >> ;
     if(T[p].l == l && T[p].r == r) {
         return T[p].Max;
     }
     if(r <= mid) {
         return query_max(p <<  , l , r);
     }
     else if(l > mid) {
         return query_max((p << )| , l , r);
     }
     else {
         return max(query_max(p <<  , l , mid) , query_max((p << )| , mid +  , r));
     }
 }

 int find_max(int u , int v) {
     int fu = top[u] , fv = top[v];
     int Max = -;
     while(fu != fv) {
         if(dep[fu] >= dep[fv]) {
             Max = max(Max , query_max( , id[fu] , id[u]));
             u = par[fu];
             fu = top[u];
         }
         else {
             Max = max(Max , query_max( , id[fv] , id[v]));
             v = par[fv];
             fv = top[v];
         }
     }
     if(dep[u] >= dep[v]) {
         return max(Max , query_max( , id[v] , id[u]));
     }
     else {
         return max(Max , query_max( , id[u] , id[v]));
     }
 }

 int find_sum(int u , int v) {
     int fu = top[u] , fv = top[v];
     int sum = ;
     while(fu != fv) {
         if(dep[fu] > dep[fv]) {
             sum += query_sum( , id[fu] , id[u]);
             u = par[fu];
             fu = top[u];
         }
         else {
             sum += query_sum( , id[fv] , id[v]);
             v = par[fv];
             fv = top[v];
         }
     }
     if(dep[u] >= dep[v]) {
         return (sum + query_sum( , id[v] , id[u]));
     }
     else {
         return (sum + query_sum( , id[u] , id[v]));
     }
 }

 int main()
 {
     int n , u , v;
     while(~scanf("%d" , &n)) {
         init();
         for(int i =  ; i < n ; ++i) {
             scanf("%d %d" , &u , &v);
             add(u , v);
             add(v , u);
         }
         for(int i =  ; i <= n ; ++i) {
             scanf("%d" , a + i);
         }
         cnt = ;
         dfs1( ,  , );
         dfs2( ,  , );
         build( ,  , cnt);
         int m;
         char q[];
         scanf("%d" , &m);
         while(m--) {
             scanf("%s%d%d" , q , &u , &v);
             if(q[] == 'C') {
                 updata( , id[u] , v);
             }
             else if(strcmp(q ,"QMAX") == ) {
                 printf("%d\n" , find_max(u , v));
             }
             else {
                 printf("%d\n", find_sum(u , v));
             }
         }
     }
     return ;
 }