ZJOI 2008 树的统计（树链剖分+线段树）

　　一棵树上有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
【题目描述】
一棵树上有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本身
【输入】
输入文件的第一行为一个整数n，表示节点的个数。
接下来n – 1行，每行2个整数a和b，表示节点a和节点b之间有一条边相连。
接下来n行，每行一个整数，第i行的整数wi表示节点i的权值。
接下来1行，为一个整数q，表示操作的总数。
接下来q行，每行一个操作，以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。
【输出】
对于每个“QMAX”或者“QSUM”的操作，每行输出一个整数表示要求输出的结果。
【输入样例】
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
【输出样例】
4
1
2
2
10
6
5
6
5
16
【数据范围】
对于100％的数据，保证1<=n<=30000，0<=q<=200000；中途操作中保证每个节点的权值w在-30000到30000之间。

在树链上 求sum求max 还是比较简单的

#pragma comment(linker,"/STACK:1024000000,1024000000")
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <vector>
using namespace std;
const int N=30010;
int n;
int tim;

int num[N],siz[N],top[N],son[N];
int dep[N],tid[N],ran[N],fa[N];
int head[N],to[2*N],nex[2*N],edge;

void init()
{
    memset(head,-1,sizeof(head));
    memset(son,-1,sizeof(son));
    tim=0;
    edge=0;
}

void addedge(int u,int v)
{
    to[edge]=v,nex[edge]=head[u],head[u]=edge++;
    to[edge]=u,nex[edge]=head[v],head[v]=edge++;
}

void dfs1(int u,int father,int d)
{

    dep[u]=d;
    fa[u]=father;
    siz[u]=1;
    for(int i=head[u];~i;i=nex[i])
    {
        int v=to[i];
        if(v!=father)
        {
            dfs1(v,u,d+1);
            siz[u]+=siz[v];
            if(son[u]==-1||siz[v]>siz[son[u]])
                son[u]=v;
        }
    }
}

void dfs2(int u,int tp)
{
    top[u]=tp;
    tid[u]=++tim;
    ran[tid[u]]=u;
    if(son[u]==-1) return ;
    dfs2(son[u],tp);
    for(int i=head[u];~i;i=nex[i])
    {
        int v=to[i];
        if(v!=son[u]&&v!=fa[u])
            dfs2(v,v);
    }
}


#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
const int inf = 0x3f3f3f3f;
int sum[4*N],mx[4*N],a[N];
void PushUP(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    mx[rt]=max(mx[rt<<1],mx[rt<<1|1]);
}

void update(int x,int y,int l,int r,int rt)
{
    if(l==r&&l==x) {
        sum[rt]=y;
        mx[rt]=y;
        return ;
    }
    int mid=(l+r)>>1;
    if(x<=mid) update(x,y,l,mid,rt<<1);
    else update(x,y,mid+1,r,rt<<1|1);
    PushUP(rt);
}

int querymx(int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r)
        return mx[rt];
    int mid=(l+r)>>1,tot=-inf;
    if(L<=mid) tot=max(tot,querymx(L,R,l,mid,rt<<1));
    if(R>mid) tot=max(tot,querymx(L,R,mid+1,r,rt<<1|1));
    return tot;
}

int querysum(int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r)
        return sum[rt];
    int mid=(l+r)>>1,tot=0;
    if(L<=mid) tot+=querysum(L,R,l,mid,rt<<1);
    if(R>mid) tot+=querysum(L,R,mid+1,r,rt<<1|1);
    return tot;
}

int solvemx(int x,int y)
{
    int maxx=-inf;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        maxx=max(maxx,querymx(tid[top[x]],tid[x],1,n,1));
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    maxx=max(maxx,querymx(tid[x],tid[y],1,n,1));
    return maxx;
}


int solvesum(int x,int y)
{
    int summ=0;
    while(top[x]!=top[y])
    {
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        summ+=querysum(tid[top[x]],tid[x],1,n,1);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    summ+=querysum(tid[x],tid[y],1,n,1);
    return summ;
}

int main()
{
    scanf("%d",&n);
    init();
    int u,v;
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        addedge(u,v);
    }
    dfs1(1,0,0);
    dfs2(1,1);

    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        update(tid[i],a[i],1,n,1);
    }
    int m;
    scanf("%d",&m);
    char ch[10];
    while(m--)
    {
        int x,y;
        scanf(" %s %d%d",ch,&x,&y);
        if(ch[0]=='C') {
            a[x]=y;update(tid[x],y,1,n,1);
        }
        else
        {
            if(ch[1]=='M') printf("%d\n",solvemx(x,y) );
            else printf("%d\n",solvesum(x,y) );
        }
    }

}