BZOJ1036--树链剖分模板

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本身

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 30006
using namespace std;
struct data{
    int l,r,max,sum,key;
}tre[maxn*4];
int n,tot,tem,tet,lnk[maxn],nxt[maxn*2],son[maxn*2],dep[maxn],H_son[maxn],top[maxn],siz[maxn],fa[maxn],id[maxn];
bool vis[maxn];
int _read(){
    int sum=0,p;char ch=getchar();
    while((!(ch>='0'&&ch<='9'))&&(ch!='-'))ch=getchar();
    if(ch=='-')p=-1,ch=getchar();else p=1;
    while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=getchar();
    return sum*p;
}
void add(int x,int y){
    nxt[++tot]=lnk[x];son[tot]=y;lnk[x]=tot;
}
void build(int p,int l,int r){
    int mid=(l+r)>>1;
    tre[p].l=l;tre[p].r=r;tre[p].max=-1e9;tre[p].key=tre[p].sum=0;
    if(l==r)return;
    build(p<<1,l,mid);build(p<<1|1,mid+1,r);
}
void update(int p,int pos,int k){
    if(pos>tre[p].r||pos<tre[p].l)return;
    if(tre[p].l==tre[p].r){
        tre[p].key=tre[p].max=tre[p].sum=k;
        return;
    }
    update(p<<1,pos,k);update(p<<1|1,pos,k);
    tre[p].max=max(tre[p<<1].max,tre[p<<1|1].max);
    tre[p].sum=tre[p<<1].sum+tre[p<<1|1].sum;
}
int query_max(int p,int l,int r){
    if(l>tre[p].r||r<tre[p].l)return -1e9;
    if(l<=tre[p].l&&r>=tre[p].r)return tre[p].max;
    return max(query_max(p<<1,l,r),query_max(p<<1|1,l,r));
}
int query_sum(int p,int l,int r){
    if(l>tre[p].r||r<tre[p].l)return 0;
    if(l<=tre[p].l&&r>=tre[p].r)return tre[p].sum;
    return query_sum(p<<1,l,r)+query_sum(p<<1|1,l,r);
}
void dfs1(int x){
    vis[x]=0;siz[x]=1;
    for(int j=lnk[x];j;j=nxt[j]) if(vis[son[j]]){
        fa[son[j]]=x;dep[son[j]]=dep[x]+1;
        dfs1(son[j]);
        if(siz[son[j]]>siz[H_son[x]])H_son[x]=son[j];
        siz[x]+=siz[son[j]];
    }
}
void dfs2(int x,int lst){
    vis[x]=0;top[x]=lst;id[x]=++tem;
    if(H_son[x])dfs2(H_son[x],lst);
    for(int j=lnk[x];j;j=nxt[j]) if(vis[son[j]]&&(son[j]!=H_son[x]))dfs2(son[j],son[j]);
}
int get_max(int x,int y){
    int Max=-1e9;
    for(;top[x]!=top[y];x=fa[top[x]]){
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        Max=max(Max,query_max(1,id[top[x]],id[x]));
    }
    if(dep[x]<dep[y])swap(x,y);
    return max(Max,query_max(1,id[y],id[x]));
}
int get_sum(int x,int y){
    int sum=0;
    for(;top[x]!=top[y];x=fa[top[x]]){
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        sum+=query_sum(1,id[top[x]],id[x]);
    }
    if(id[x]>id[y])swap(x,y);
    return sum+query_sum(1,id[x],id[y]);
}
int main(){
    freopen("count.in","r",stdin);
    freopen("count.out","w",stdout);
    n=_read();
    for(int i=1,x,y;i<n;i++)x=_read(),y=_read(),add(x,y),add(y,x);
    memset(vis,1,sizeof(vis));
    dfs1(1);
    memset(vis,1,sizeof(vis));
    dfs2(1,1);
    build(1,1,tem);
    for(int i=1,x;i<=n;i++)x=_read(),update(1,id[i],x);
    tet=_read();
    while(tet--){
        char s[30];
        scanf("%s",s);int x=_read(),y=_read();
        if(s[0]=='C')update(1,id[x],y);else
        if(s[1]=='M')printf("%d\n",get_max(x,y));else
        if(s[1]=='S')printf("%d\n",get_sum(x,y));
    }
    return 0;
}