树链剖分点更新模板HYSBZ 1036树的统计Count

Description

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

4
1
2
2
10
6
5
6
5
16

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <queue>
#include <iostream>
#include <algorithm>
#define lson l,mid,o<<1
#define rson mid+1,r,o<<1|1
using namespace std;
const int maxn = 30005;
int n;
int son[maxn]; //表示该点重链的子节点  
int num[maxn]; //表示以该点为根的子树的节点数量  
int dep[maxn]; //表示该节点的深度  
int fa[maxn];  // 表示该节点的父亲节点  
int vis[maxn]; // 表示该节点在线段树中的编号  
int top[maxn]; // 表示这条重链的顶端节点  
int now;  
int head[maxn],cnt,A,B,C;
struct Edge
{
    int v,next;
}e[maxn<<1];
void add(int a,int b)
{
    e[cnt].v = b;
    e[cnt].next = head[a];
    head[a] = cnt++;
}
void init()
{
    memset(head,-1,sizeof(head));
    memset(son,-1,sizeof(son));   
    now = cnt = 0;
}
void dfs1(int u,int v,int step) //第一遍dfs求出son，fa，num，dep  
{
    num[v] = 1;dep[v] = step;fa[v] = u;
    int i;
    for(i = head[v]; i != -1; i = e[i].next)
    {
        int end = e[i].v;
        if(end == u)
            continue;
        dfs1(v,end,step+1);
        if(son[v] == -1 || num[son[v]] < num[end])// 取子节点数量最多的点  
        {
            son[v] = end;
        }
        num[v] += num[end];
    }
}
void dfs2(int u,int v)// 第二遍dfs求出top和vis  
{
    top[v] = u;
    vis[v] = ++now; // 一个节点在线段树里面代表一个点
    if(son[v] != -1)  //将重链在线段树里连续编号  
    {
        dfs2(u,son[v]);
    }
    else
        return;
    int i;
    for(i = head[v] ; i != -1; i = e[i].next)
    {
        int end = e[i].v;
        if(end == fa[v] || end == son[v])//等于父亲节点或者等于重链的子节点跳过，这里的u不是父亲节点，是重链的顶端 
            continue;
        dfs2(end,end); // 所有轻链重新为其节点的重链编号  
    }
}
int sum[maxn<<2],ma[maxn<<2];
void update(int l,int r,int o)
{
    if(l == r)
    {
        sum[o] = ma[o] = B;
        return;
    }
    int mid = (l + r) >> 1;
    if(A <= mid) update(lson);
    else update(rson);
    sum[o] = sum[o<<1] + sum[o << 1 | 1];
    ma[o] = max(ma[o<<1],ma[o<<1|1]);
}
int ansma,anss;
void qurry(int l,int r,int o)
{
    if(A <= l && r <= B)
    {
        anss += sum[o];
        ansma = max(ansma,ma[o]);
        return;
    }
    int mid = (l + r) >> 1;
    if(A <= mid) qurry(lson);
    if(mid < B) qurry(rson);
}
//更新的时候记得用vis取得线段树下标，不要用原来树中的编号。
void solve(int a,int b,int c)
{
    if(c == 0)
    {
        A = vis[a];
        B = b;
        update(1,n,1);
    }
    else
    {
        int q = top[a],w = top[b];
        ansma = -1000005,anss = 0;
        while(q != w)
        {
            if(dep[q] < dep[w])
            {
                swap(q,w);
                swap(a,b);
            }
            A = vis[q] , B = vis[a];
            qurry(1,n,1);
            //记得更新a跟q顺序不能反
            a = fa[q];
            q = top[a];
        }
        if(dep[a] > dep[b])
        {
            swap(a,b);
        }
        A = vis[a],B = vis[b];
        qurry(1,n,1);
        if(c == 1)
            printf("%d\n",ansma);
        else
            printf("%d\n",anss);
    }
}
int main()
{
    while(scanf("%d",&n) != EOF)
    {
        init();
        int i,j,a1,b1;
        for(i = 1; i < n; i++)
        {
            scanf("%d %d",&a1,&b1);
            add(a1,b1);
            add(b1,a1);
        }
        dfs1(1,1,1);
        dfs2(1,1);
        for(i = 1; i <= n; i++)
        {
            scanf("%d",&a1);
            A = vis[i];B = a1;
            update(1,n,1);
        }
        int m;
        scanf("%d",&m);
        char str[10];
        while(m--)
        {
            scanf("%s %d %d",str,&a1,&b1);
            int now = 0;
            if(str[0] == 'Q' && str[1] == 'M')
                now = 1;
            else if(str[0] == 'Q' && str[1] == 'S')
                now = 2;
            solve(a1,b1,now);
        }
    }
    return 0;
}