[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[2], rev, val, sum, max;
    inline int& operator[] (int rhs)
    {
        return c[rhs];
    }
}a[30005];
int sta[30005], top;
 
void push_up(int k)
{
    a[k].sum = a[a[k][0]].sum + a[a[k][1]].sum + a[k].val;
    a[k].max = max(max(a[a[k][0]].max, a[a[k][1]].max), a[k].val);
}
 
void push_down(int k)
{
    if(a[k].rev)
    {
        a[a[k][0]].rev ^= 1, a[a[k][1]].rev ^= 1;
        swap(a[k][0], a[k][1]), a[k].rev = 0;
    }
}
 
bool isroot(int x)
{
    return a[a[x].fa][0] != x && a[a[x].fa][1] != x;
}
 
void rotate(int x)
{
    int y = a[x].fa, z = a[y].fa;
    int dy = a[y][1] == x, dz = a[z][1] == y;
    if(!isroot(y)) a[z][dz] = x;
    a[y][dy] = a[x][dy ^ 1], a[a[x][dy ^ 1]].fa = y;
    a[x][dy ^ 1] = y, a[y].fa = x, a[x].fa = z;
    push_up(y);
}
 
void splay(int x)
{
    sta[top = 1] = 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][1] == x ^ a[z][1] == y) rotate(x);
            else rotate(y);
        rotate(x);
    }
    push_up(x);
}
 
void access(int x)
{
    for(int i = 0; x; x = a[x].fa)
        splay(x), a[x][1] = i, i = x;
}
 
void make_root(int x)
{
    access(x), splay(x), a[x].rev ^= 1;
}
 
int main()
{
    int n, q, u, v;
    char s[10];
    scanf("%d", &n), a[0].max = -33333;
    for(int i = 1; i < n; i++)
        scanf("%d%d", &u, &v), make_root(u), a[u].fa = v;
    for(int i = 1; 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[1] == 'H')
            splay(u), a[u].val = v, push_up(u);
        if(s[1] == 'M')
        {
            make_root(u), access(v), splay(v);
            printf("%d\n", a[v].max);
        }
        if(s[1] == 'S')
        {
            make_root(u), access(v), splay(v);
            printf("%d\n", a[v].sum);
        }
    }
    return 0;
}