[BZOJ1036] [ZJOI2008] 树的统计Count (LCT)

Description

　　一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成
一些操作： I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I
II. QSUM u v: 询问从点u到点v的路径上的节点的权值和 注意：从点u到点v的路径上的节点包括u和v本身

Input

　　输入的第一行为一个整数n，表示节点的个数。接下来n – 1行，每行2个整数a和b，表示节点a和节点b之间有
一条边相连。接下来n行，每行一个整数，第i行的整数wi表示节点i的权值。接下来1行，为一个整数q，表示操作
的总数。接下来q行，每行一个操作，以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。 
　　对于100％的数据，保证1<=n<=30000，0<=q<=200000；中途操作中保证每个节点的权值w在-30000到30000之间。

Output

　　对于每个“QMAX”或者“QSUM”的操作，每行输出一个整数表示要求输出的结果。

Sample Input

4
1 2
2 3
4 1
4 2 1 3
12
QMAX 3 4
QMAX 3 3
QMAX 3 2
QMAX 2 3
QSUM 3 4
QSUM 2 1
CHANGE 1 5
QMAX 3 4
CHANGE 3 6
QMAX 3 4
QMAX 2 4
QSUM 3 4

Sample Output

4
1
2
2
10
6
5
6
5
16

HINT

Source

Solution

　　LCT维护点权和与点权最大值。注意用cin/cout会RE！！！

 #include <bits/stdc++.h>
 using namespace std;
 struct LCT
 {
     int fa, c[], rev, val, sum, max;
     inline int& operator[] (int rhs)
     {
         return c[rhs];
     }
 }a[];
 int sta[], top;

 void push_up(int k)
 {
     a[k].sum = a[a[k][]].sum + a[a[k][]].sum + a[k].val;
     a[k].max = max(max(a[a[k][]].max, a[a[k][]].max), a[k].val);
 }

 void push_down(int k)
 {
     if(a[k].rev)
     {
         a[a[k][]].rev ^= , a[a[k][]].rev ^= ;
         swap(a[k][], a[k][]), a[k].rev = ;
     }
 }

 bool isroot(int x)
 {
     return a[a[x].fa][] != x && a[a[x].fa][] != x;
 }

 void rotate(int x)
 {
     int y = a[x].fa, z = a[y].fa;
     int dy = a[y][] == x, dz = a[z][] == y;
     if(!isroot(y)) a[z][dz] = x;
     a[y][dy] = a[x][dy ^ ], a[a[x][dy ^ ]].fa = y;
     a[x][dy ^ ] = y, a[y].fa = x, a[x].fa = z;
     push_up(y);
 }

 void splay(int x)
 {
     sta[top = ] = x;
     for(int i = x; !isroot(i); i = a[i].fa)
         sta[++top] = a[i].fa;
     while(top)
         push_down(sta[top--]);
     while(!isroot(x))
     {
         int y = a[x].fa, z = a[y].fa;
         if(!isroot(y))
             if(a[y][] == x ^ a[z][] == y) rotate(x);
             else rotate(y);
         rotate(x);
     }
     push_up(x);
 }

 void access(int x)
 {
     for(int i = ; x; x = a[x].fa)
         splay(x), a[x][] = i, i = x;
 }

 void make_root(int x)
 {
     access(x), splay(x), a[x].rev ^= ;
 }

 int main()
 {
     int n, q, u, v;
     char s[];
     scanf("%d", &n), a[].max = -;
     for(int i = ; i < n; i++)
         scanf("%d%d", &u, &v), make_root(u), a[u].fa = v;
     for(int i = ; i <= n; i++)
         splay(i), scanf("%d", &a[i].val), push_up(i);
     scanf("%d", &q);
     while(q--)
     {
         scanf("%s%d%d", s, &u, &v);
         if(s[] == 'H')
             splay(u), a[u].val = v, push_up(u);
         if(s[] == 'M')
         {
             make_root(u), access(v), splay(v);
             printf("%d\n", a[v].max);
         }
         if(s[] == 'S')
         {
             make_root(u), access(v), splay(v);
             printf("%d\n", a[v].sum);
         }
     }
     return ;
 }