[BZOJ]1036: [ZJOI2008]树的统计Count

一道树链剖分的裸题，要注意找最大值的时候ans要设得很小，我因此WA了一次。。。。。。

/**************************************************************
    Problem: 1036
    User: 200815147
    Language: C++
    Result: Accepted
    Time:2792 ms
    Memory:7816 kb
****************************************************************/
 
#include<cstdio>
#include<cstring>
const int Q=50005;
struct edge
{
    int y,next;
}b[Q*2];
int len=0,last[Q];
void ins(int x,int y)
{
    int t=++len;
    b[t].y=y;b[t].next=last[x];last[x]=t;
}
int tot[Q],son[Q],dep[Q],ys[Q],fa[Q],top[Q];
void pre_node(int x)
{
    tot[x]=1;son[x]=0;
    for(int i=last[x];i!=-1;i=b[i].next)
    {
        int y=b[i].y;
        if(fa[x]!=y)
        {
            fa[y]=x;
            dep[y]=dep[x]+1;
            pre_node(y);
            if(tot[y]>tot[son[x]]) son[x]=y;
            tot[x]+=tot[y];
        }
    }
}
int z=0;
void pre_edge(int x,int tp)
{
    ys[x]=++z;top[x]=tp;
    if(son[x]!=0) pre_edge(son[x],tp);
    for(int i=last[x];i!=-1;i=b[i].next)
    {
        int y=b[i].y;
        if(son[x]!=y && fa[x]!=y) pre_edge(y,y);
    }
}
struct tree
{
    int l,r,Max,Sum,lc,rc;
}a[Q*3];
int trlen=0;
void build(int l,int r)
{
    int t=++trlen;
    a[t].l=l;a[t].r=r;
    if(l<r)
    {
        int mid=(l+r)>>1;
        a[t].lc=trlen+1;build(l,mid);
        a[t].rc=trlen+1;build(mid+1,r);
    }
}
int mymax(int x,int y) {return x>y?x:y;}
void change(int now,int x,int c)
{
    if(a[now].l==a[now].r)
    {
        a[now].Max=a[now].Sum=c;
        return;
    }
    int lc=a[now].lc,rc=a[now].rc,mid=(a[now].l+a[now].r)>>1;
    if(x<=mid) change(lc,x,c);
    else change(rc,x,c);
    a[now].Sum=a[lc].Sum+a[rc].Sum;
    a[now].Max=mymax(a[lc].Max,a[rc].Max);
}
int findmax(int now,int l,int r)
{
    if(a[now].l==l && a[now].r==r) return a[now].Max;
    int lc=a[now].lc,rc=a[now].rc,mid=(a[now].l+a[now].r)>>1;
    if(r<=mid) return findmax(lc,l,r);
    else if(l>mid) return findmax(rc,l,r);
    else return mymax(findmax(lc,l,mid),findmax(rc,mid+1,r));
}
int findsum(int now,int l,int r)
{
    if(a[now].l==l && a[now].r==r) return a[now].Sum;
    int lc=a[now].lc,rc=a[now].rc,mid=(a[now].l+a[now].r)>>1;
    if(r<=mid) return findsum(lc,l,r);
    else if(l>mid) return findsum(rc,l,r);
    else return findsum(lc,l,mid)+findsum(rc,mid+1,r);
}
int n,m;
void solve1(int x,int y)
{
    int tx=top[x],ty=top[y],ans=-300000;
    while(tx!=ty)
    {
        //printf("x=%d y=%d tx=%d ty=%d\n",x,y,tx,ty);
        if(dep[tx]>dep[ty])
        {
            int t=tx;tx=ty;ty=t;
            t=x;x=y;y=t;
        }
        //printf("x=%d y=%d tx=%d ty=%d\n",x,y,tx,ty);
        ans=mymax(ans,findmax(1,ys[ty],ys[y]));
        //printf("%d %d %d\n",ys[ty],ys[y],ans);
        y=fa[ty];ty=top[y];
    }
    if(dep[x]>dep[y]) {int t=x;x=y;y=t;}
    ans=mymax(ans,findmax(1,ys[x],ys[y]));
    printf("%d\n",ans);
}
void solve2(int x,int y)
{
    int tx=top[x],ty=top[y],ans=0;
    while(tx!=ty)
    {
        if(dep[tx]>dep[ty])
        {
            int t=tx;tx=ty;ty=t;
            t=x;x=y;y=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;}
    /*printf("%d %d %d %d\n",x,y,ys[x],ys[y]);
    printf("%d\n",ans);*/
    ans+=findsum(1,ys[x],ys[y]);
    printf("%d\n",ans);
}
int main()
{
    //freopen("1.txt","w",stdout);
    memset(last,-1,sizeof(last));
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        ins(x,y);ins(y,x);
    }//printf("ok");
    tot[0]=dep[1]=fa[1]=0;pre_node(1);
    pre_edge(1,1);
    build(1,n);
    for(int i=1;i<=n;i++)
    {
        int d;
        scanf("%d",&d);
        change(1,ys[i],d);
    }
    scanf("%d",&m);//for(int i=1;i<=len;i++)
    //printf("%d %d %d %d\n",i,a[i].l,a[i].r,a[i].Sum);
    for(int i=1;i<=m;i++)
    {
        char s[10];
        int x,y;
        scanf("%s%d%d",s,&x,&y);
        if(s[0]=='C') change(1,ys[x],y);
        else if(s[1]=='M') solve1(x,y);
        else solve2(x,y);
    }
}