bzoj1036[ZJOI2008]树的统计Count

这道题裸裸的树剖。。
求个和求个最大值即可。。
不写什么了，不会树剖的自己学。。

#include<cstdio>
#include<cstring>
using namespace std;
struct node{
    int x,y,next;
}a[2100000];int len,last[110000]; 
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 lc,rc,l,r,c,cc;
}tr[410000];int trlen;
void bt(int l,int r){
    trlen++; int now=trlen;
    tr[now].l=l;tr[now].r=r; tr[now].c=tr[now].cc=-1;
    tr[now].lc=tr[now].rc=-1;
    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 n,fa[110000],dep[110000],son[110000],tot[110000];
void pre_tree_node(int x){
    tot[x]=1;son[x]=-1;
    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]==-1)  
                son[x]=y;
            tot[x]+=tot[y];   
        }   
    }
}
int z,ys[110000],top[110000];
void pre_tree_edge(int x,int tp){
    ys[x]=++z;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!=son[x] && y!=fa[x] )
            pre_tree_edge(y,y); 
    }

}
int mymax(int x,int y) { 
    return x>y?x:y;
}
void change(int now,int p,int c) {
    if( tr[now].l==tr[now].r) { 
        tr[now].c=c;tr[now].cc=c;return ;
    }
    int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
    if( p<=mid) 
        change(lc,p,c);
    else        
        change(rc,p,c);
    tr[now].c=mymax( tr[lc].c,tr[rc].c);
    tr[now].cc=tr[lc].cc+tr[rc].cc;
}
int findmax(int now,int l,int r){
    if( l== tr[now].l && tr[now].r==r) 
        return tr[now].c;
    int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
    if(mid+1<=l)        
        return findmax(rc,l,r);  
    else if( r<=mid) 
        return findmax(lc,l,r);  
    else             
        return mymax(findmax(lc,l,mid),findmax(rc,mid+1,r));
}
int findsum(int now,int l,int r){
    if(l==tr[now].l && tr[now].r==r) 
        return tr[now].cc;
    int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
    if(mid+1<=l)        
        return findsum(rc,l,r);  
    else if( r<=mid) 
        return findsum(lc,l,r);  
    else             
        return findsum(lc,l,mid)+findsum(rc,mid+1,r);
}
int solve1(int x,int y){
    int tx=top[x],ty=top[y],ans=-999999;
    while(tx!=ty){
        if(dep[ tx ]>dep[ ty ] ) {                                                
            int t=x;x=y;y=t; 
            t=tx;tx=ty;ty=t;
        }
        ans=mymax( ans, findmax(1,ys[ty],ys[ y]) );
        y=fa[ty];ty=top[y];
    }
    if(dep[x]>dep[y]){ 
        int t=x;x=y;y=t; 
    }
    return mymax( ans,findmax(1,ys[x], ys[y]) );
}
int solve2(int x,int y){
    int tx=top[x],ty=top[y],ans=0;
    while(tx!=ty){
        if(dep[tx]>dep[ty]) {                                                
            int t=x;x=y;y=t; 
            t=tx;tx=ty;ty=t;
        }
        ans+=findsum(1,ys[ty],ys[y] );
        y=fa[ty];ty=top[y];
    } 
    if(dep[x]>dep[y]){ 
        int t=x;x=y;y=t; 
    }
    return ans+findsum(1,ys[x],ys[y]);
}
struct bian{
    int x,y,c;
}e[110000]; 
int d[31000];
int main(){
    while(scanf("%d",&n)!=EOF) {
        int m;
        len=0;memset(last,0,sizeof(last));
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&e[i].x,&e[i].y);
            if(e[i].x>e[i].y) {
                int t=e[i].x;e[i].x=e[i].y;e[i].y=t;
            }
            ins(e[i].x,e[i].y);
            ins(e[i].y,e[i].x);
        }
        for(int i=1;i<=n;i++)
            scanf("%d",&d[i]);
        scanf("%d",&m);
        dep[1]=0;fa[1]=0;pre_tree_node(1);
        z=0;pre_tree_edge(1,1);
        trlen=0; bt(1,z); 
        for(int i=1;i<=n;i++) 
            change(1,ys[i],d[i]);
        for(int i=1;i<=m;i++) {
            char ss[21];scanf("%s",ss+1);
            int x,y;scanf("%d%d",&x,&y);
            if(ss[1]=='C') {
                change(1,ys[x],y);
            }
            else {

                if(ss[2]=='M')
                    printf("%d\n",solve1(x,y));
                else
                    printf("%d\n",solve2(x,y));
            }
        }
    }
    return 0;
}