[BZOJ1036][ZJOI2008]树的统计Count

线段树 树链剖分╮(￣▽￣)╭

传送门

---

Descriptionヾ(o´∀｀o)ﾉ

一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成一些操作：
I. CHANGE u t : 把结点u的权值改为t
II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I 
III. 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ヾ(๑╹◡╹)ﾉ”

1

2

2

10

6

5

6

5

16

Solution(⁎˃ᴗ˂⁎)

树链剖分的模板题 用线段树来进行维护

不懂树链剖分的戳这里

Code٩(๑>◡<๑)۶

#include <cstdio>
#include <cstring>  
#include <cstdlib> 
#include <iostream> 
#include <algorithm>  
#define N 30010
using namespace std;
struct tree
{
    int l, r, sum, maxx;
} t[150010];
int cnt, sz, n, q, fa[N], pos[N], dep[N], size[N], head[N], to[N + N], next[N + N], w[N], belong[N];
void add(int u, int v)
{
    to[++cnt] = v;
    next[cnt] = head[u];
    head[u] = cnt;
}
void dfs1(int x)
{
    size[x] = 1;
    for (int i = head[x]; i; i = next[i])
    {
        if (to[i] == fa[x])
            continue;
        dep[to[i]] = dep[x] + 1;
        fa[to[i]] = x;
        dfs1(to[i]);
        size[x] += size[to[i]];
    }
}
void dfs2(int x, int chain)
{
    int k = 0; sz++;
    pos[x] = sz;
    belong[x] = chain;
    for (int i = head[x]; i; i = next[i])
        if (dep[to[i]] > dep[x] && size[to[i]] > size[k])
            k = to[i];
    if (k == 0)
        return;
    dfs2(k, chain);
    for (int i = head[x]; i; i = next[i])
        if (dep[to[i]] > dep[x] && k != to[i])
            dfs2(to[i], to[i]);
}
void build(int x, int l, int r) //建线段树
{
    t[x].l = l;
    t[x].r = r;
    if (l == r)
        return;
    int mid = (l + r ) >> 1;
    build(x << 1, l, mid);
    build(x << 1 | 1, mid + 1, r);
}
void changeone(int x, int xx, int  yy) //单点修改 
{
    int l = t[x].l, r = t[x].r, mid = (l + r) >> 1;
    if (l == r)
    {
        t[x].sum = t[x].maxx = yy;
        return;
    }
    if (xx <= mid)
        changeone(x << 1, xx, yy);
    else 
        changeone(x << 1 | 1, xx, yy);
    t[x].sum = t[x << 1].sum + t[x << 1 | 1].sum;
    t[x].maxx = max(t[x << 1].maxx, t[x << 1 | 1].maxx);
}
int querysum(int x, int xx, int yy) //区间求和 
{
    int l = t[x].l, r = t[x].r, mid = (l + r) >> 1;
    if (l == xx && r == yy)
        return t[x].sum;
    if (yy <= mid)
        return querysum(x << 1, xx, yy);
    else
        if (xx > mid)
            return querysum(x << 1 | 1, xx, yy);
        else
            return querysum(x << 1, xx, mid) + querysum(x << 1 | 1, mid + 1, yy);
}
int querymax(int x,int xx,int yy) //区间求Max 
{
    int l = t[x].l, r = t[x].r, mid = (l + r) >> 1;
    if (l == xx && r == yy)
        return t[x].maxx;
    if (yy <= mid)
        return querymax(x << 1, xx, yy); 
    else
        if (xx > mid)
            return querymax(x << 1 | 1, xx, yy);
        else
            return max(querymax(x << 1, xx, mid), querymax(x << 1 | 1, mid + 1, yy));
}
int anssum(int x, int y)
{
    int sum = 0;
    while (belong[x] != belong[y])
    {
        if (dep[belong[x]] < dep[belong[y]])
            swap(x, y); 
        sum += querysum(1, pos[belong[x]], pos[x]);
        x = fa[belong[x]];
    }
    if (pos[x] > pos[y])
        swap(x, y);
    sum += querysum(1, pos[x], pos[y]);
    return sum;
}
int ansmax(int x, int y)
{
    int mx = -10000000;
    while (belong[x] != belong[y])
    {
        if (dep[belong[x]] < dep[belong[y]])
            swap(x, y); 
        mx = max(mx, querymax(1, pos[belong[x]], pos[x]));
        x = fa[belong[x]];
    }
    if (pos[x] > pos[y])
        swap(x, y);
    mx = max(mx, querymax(1, pos[x], pos[y]));
    return mx;
}
int main()  
{
    cnt = 0;
    scanf("%d", &n);
    for (int i = 1; i < n; i++)
    {
        int u, v;
        scanf("%d%d", &u, &v);
        add(u, v);
        add(v, u);
    }
    for (int i = 1; i <= n; i++)
        scanf("%d", &w[i]);

    dfs1(1);
    dfs2(1, 1); 
    build(1, 1, n);
    for (int i = 1; i <= n; i++)
        changeone(1, pos[i], w[i]);

    scanf("%d", &q);
    char ch[7];
    for (int i = 1; i <= q; i++) 
    {
        int u, v;
        scanf("%s%d%d", ch, &u, &v);
        if (ch[0] == 'C')
        {
            w[u] = v;
            changeone(1, pos[u], v);
        }
        else
        {
            if (ch[1] == 'M')         
                printf("%d\n", ansmax(u, v));
            else
                printf("%d\n", anssum(u, v));
        }
    }
    return 0;
}