[bzoj 1036--ZJOI2008]树的统计

一棵树上有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本身

bzoj中难得的模板题啊！日常做水题真是哈皮。树剖裸题。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
struct node
{
    int x,y,next;
}a[610000];int len,last[310000];
void ins(int x,int y)
{
    len++;
    a[len].x=x;a[len].y=y;
    a[len].next=last[x];last[x]=len;
}
struct trnode
{
    int l,r,lc,rc,mx,sum;
}tr[610000];int trlen;
void bt(int l,int r)
{
    trlen++;int now=trlen;
    tr[now].l=l;tr[now].r=r;
    tr[now].lc=tr[now].rc=-1;
    tr[now].mx=-99999;tr[now].sum=0;
    if(l<r)
    {
        int mid=(l+r)/2;
        tr[now].lc=trlen+1;bt(l,mid);
        tr[now].rc=trlen+1;bt(mid+1,r);
    }
}
int fa[31000],dep[31000],tot[31000],son[31000];
void pre_tree_node(int x)
{
    tot[x]=1;son[x]=0;
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa[x])
        {
            fa[y]=x;
            dep[y]=dep[x]+1;
            pre_tree_node(y);
            if(tot[son[x]]<tot[y])son[x]=y;
            tot[x]+=tot[y];
        }
    }
}
int id,ys[31000],top[31000];
void pre_tree_edge(int x,int tp)
{
    ys[x]=++id;top[x]=tp;
    if(son[x]!=0)pre_tree_edge(son[x],tp);
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa[x] && y!=son[x])pre_tree_edge(y,y);
    }
}
void change(int now,int x,int k)
{
    if(tr[now].l==tr[now].r){tr[now].mx=tr[now].sum=k;return ;}
    int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
    if(x<=mid)change(lc,x,k);
    else change(rc,x,k);
    tr[now].mx=max(tr[lc].mx,tr[rc].mx);
    tr[now].sum=tr[lc].sum+tr[rc].sum;
}
int findmax(int now,int l,int r)
{
    if(tr[now].l==l && tr[now].r==r)return tr[now].mx;
    int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
    if(r<=mid)return findmax(lc,l,r);
    else if(mid+1<=l)return findmax(rc,l,r);
    else return max(findmax(lc,l,mid),findmax(rc,mid+1,r));
}
int getsum(int now,int l,int r)
{
    if(tr[now].l==l && tr[now].r==r)return tr[now].sum;
    int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
    if(r<=mid)return getsum(lc,l,r);
    else if(mid+1<=l)return getsum(rc,l,r);
    else return getsum(lc,l,mid)+getsum(rc,mid+1,r);
}
int solve1(int x,int y)
{
    int tx=top[x],ty=top[y],ans=-99999;
    while(tx!=ty)
    {
        if(dep[tx]>dep[ty])swap(x,y),swap(tx,ty);
        ans=max(ans,findmax(1,ys[ty],ys[y]));
        y=fa[ty];ty=top[y];
    }
    if(dep[x]>dep[y])swap(x,y);
    ans=max(ans,findmax(1,ys[x],ys[y]));
    return ans;
}
int solve2(int x,int y)
{
    int tx=top[x],ty=top[y],ans=0;
    while(tx!=ty)
    {
        if(dep[tx]>dep[ty])swap(x,y),swap(tx,ty);
        ans+=getsum(1,ys[ty],ys[y]);
        y=fa[ty];ty=top[y];
    }
    if(dep[x]>dep[y])swap(x,y);
    ans+=getsum(1,ys[x],ys[y]);
    return ans;
}
char s[10];
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        ins(x,y);ins(y,x);
    }
    pre_tree_node(1);pre_tree_edge(1,1);bt(1,n);
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        change(1,ys[i],x);
    }
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int x,y;
        scanf("%s%d%d",s+1,&x,&y);
        if(s[2]=='H')change(1,ys[x],y);
        if(s[2]=='M')printf("%d\n",solve1(x,y));
        if(s[2]=='S')printf("%d\n",solve2(x,y));
    }
    return 0;
}