BZOJ-1036: [ZJOI2008]树的统计Count (树链剖分 线段树 单点修改 区间查询 入门题)

1036: [ZJOI2008]树的统计Count

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 18897   Solved: 7713
[ Submit][ Status][ Discuss]

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

#include <bits/stdc++.h>
using namespace std;
#define maxn 30005
vector<vector<int> >g(maxn);
int pre[maxn], sz[maxn], son[maxn], top[maxn], dep[maxn], id[maxn], val[maxn], tot;
struct tree{
    int mx, sm;
}c[maxn * 4];
void dfs1(int x, int fa, int d){
    dep[x] = d;
    sz[x] = 1;
    pre[x] = fa;
    son[x] = 0;
    int cur;
    for(int i = 0; i < g[x].size(); ++i){
        cur = g[x][i];
        if(cur == fa) continue;
        dfs1(cur, x, d + 1);
        sz[x] += sz[cur];
        if(sz[son[x]] < sz[cur]){
            son[x] = cur;
        }
    }
}
void dfs2(int x, int tp){
    top[x] = tp;
    id[x] = ++tot;
    if(son[x]){
        dfs2(son[x], tp);
    }
    int cur;
    for(int i = 0; i < g[x].size(); ++i){
        cur = g[x][i];
        if(cur == pre[x] || cur == son[x]) continue;
        dfs2(cur, cur);
    }
}
void build(int o, int l, int r){
    if(l == r){
        c[o].mx = val[l];
        c[o].sm = val[l];
        return;
    }
    int mid = l + r >> 1;
    build(o << 1 | 1, mid + 1, r);
    build(o << 1, l, mid);
    c[o].sm = c[o << 1].sm + c[o << 1 | 1].sm;
    c[o].mx = max(c[o << 1].mx, c[o << 1 | 1].mx);
}
int query(int o, int l, int r, int L, int R, int op){
    if(l >= L && r <= R){
        if(op == 1){
            return c[o].mx;
        }
        else{
            return c[o].sm;
        }
    }
    int mid = l + r >> 1;
    int ans;
    if(op == 1){
        ans = -0x3f3f3f3f;
        if(mid < R) ans = max(ans, query(o << 1 | 1, mid + 1, r, L, R, 1));
        if(mid >= L) ans = max(ans, query(o << 1, l, mid, L, R, 1));
    }
    else{
        ans = 0;
        if(mid < R) ans += query(o << 1 | 1, mid + 1, r, L, R, 2);
        if(mid >= L) ans += query(o << 1, l, mid, L, R, 2);
    }
    return ans;
}
int getmx(int x, int y){
    int tp1 = top[x], tp2 = top[y];
    int ans = -0x3f3f3f3f;
    while(tp1 != tp2){
        if(dep[tp1] < dep[tp2]){
            int temp = tp1; tp1 = tp2; tp2 = temp;
            temp = x; x = y; y = temp;
        }
        ans = max(ans, query(1, 1, tot, id[tp1], id[x], 1));
        x = pre[tp1];
        tp1 = top[x];
    }
    if(dep[x] < dep[y]){
        int temp = x; x = y; y = temp;
    }
    ans = max(ans, query(1, 1, tot, id[y], id[x], 1));
    return ans;
}
int getsm(int x, int y){
    int tp1 = top[x], tp2 = top[y];
    int ans = 0;
    while(tp1 != tp2){
        if(dep[tp1] < dep[tp2]){
            int temp = tp1; tp1 = tp2; tp2 = temp;
            temp = x; x = y; y = temp;
        }
        ans += query(1, 1, tot, id[tp1], id[x], 2);
        x = pre[tp1];
        tp1 = top[x];
    }
    if(dep[x] < dep[y]){
        int temp = x; x = y; y = temp;
    }
    ans += query(1, 1, tot, id[y], id[x], 2);
    return ans;
}
void update(int o, int l, int r, int pos, int v){
    if(l == r){
        c[o].mx = c[o].sm = v;
        return;
    }
    int mid = l + r >> 1;
    if(pos <= mid) update(o << 1, l, mid, pos, v);
    else update(o << 1 | 1, mid + 1, r, pos, v);
    c[o].mx = max(c[o << 1].mx, c[o << 1 | 1].mx);
    c[o].sm = c[o << 1].sm + c[o << 1 | 1].sm;
}
int main(){
    tot = 0;
    int n;
    scanf("%d", &n);
    int x, y;
    for(int i = 1; i < n; ++i){
        scanf("%d %d", &x, &y);
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs1(1, 0, 1);
    dfs2(1, 1);     
    for(int i = 1; i <= n; ++i){
         scanf("%d", &val[id[i]]);
    }
    build(1, 1, tot);
    char s[10];
    int q;
    scanf("%d", &q);
    for(int i = 1; i <= q; ++i){
        scanf("%s", s);
        if(s[1] == 'M'){
            scanf("%d %d", &x, &y);
            printf("%d\n", getmx(x, y));
        }
        else if(s[1] == 'S'){
            scanf("%d %d", &x, &y);
            printf("%d\n", getsm(x, y));
        }
        else{
            scanf("%d %d", &x, &y);
            update(1, 1, tot, id[x], y);
        }
    }
}

/*
题意：一棵树3e4个点，每个点有权值。2e5次操作，要么修改一个点上的值，要么询问两点之间的最大值或权值和。

思路：树链剖分的入门题。暴力的话，就是先找到LCA，然后从两个起始点一个个的往上统计点上的值。树链剖分的作用就是，在
你往上枚举点的时候，不用一个个的去枚举，而是成段的去统计，这样可以让枚举的效率更高。而这些成段的区间在线段树上是
可以连续且一一对应的。所以这题就是分两个部分，一个是用树链剖分把整棵数剖分成一些区间，然后对应到线段树上，然后
在线段树上做单点修改，区间查询的操作，那么线段树部分属于简单的应用了。
*/