【树链剖分】[ZJOI2008]树的统计Count

题目
树链剖分，再存代码。

#include<cstdio>
#include<algorithm>
#define MAXN 30000
#define MAXLOG 15
#define INF 0x7fffffff
using namespace std;
int m,n,size[MAXN+10],dep[MAXN+10],f[MAXN+10][MAXLOG+1],bl[MAXN+10],pos[MAXN+10],wt[MAXN+10],length[MAXN+10];
struct node{
    int v;
    node *next;
}*adj[MAXN+10],edge[MAXN*2+10],*ecnt=edge;
struct seg_tree{
    int mx,sum;
}tree[MAXN*4+10],*ttop=tree,*root[MAXN+10];
void addedge(int u,int v){
    node *p=++ecnt;
    p->v=v;
    p->next=adj[u];
    adj[u]=p;
}
void Read(int &x){
    char c;
    bool f=0;
    while(c=getchar(),c!=EOF){
        if(c=='-')
            f=1;
        else if(c>='0'&&c<='9'){
            x=c-'0';
            while(c=getchar(),c>='0'&&c<='9')
                x=x*10+c-'0';
            ungetc(c,stdin);
            if(f)
                x=-x;
            return;
        }
    }
}
void dfs1(int u){
    size[u]=1;
    for(int i=1;i<=MAXLOG;i++)
        f[u][i]=f[f[u][i-1]][i-1];
    for(node *p=adj[u];p;p=p->next)
        if(p->v!=f[u][0]){
            f[p->v][0]=u;
            dep[p->v]=dep[u]+1;
            dfs1(p->v);
            size[u]+=size[p->v];
        }
}
inline seg_tree *NewTree(int len){
    seg_tree *ret=ttop;
    ttop+=4*len;
    return ret;
}
void insert(seg_tree *tree,int i,int l,int r,int pos,int wt){
    if(l==r){
        tree[i].mx=tree[i].sum=wt;
        return;
    }
    int mid=(l+r)>>1;
    if(pos<=mid)
        insert(tree,i<<1,l,mid,pos,wt);
    else
        insert(tree,(i<<1)|1,mid+1,r,pos,wt);
    tree[i].mx=max(tree[i<<1].mx,tree[(i<<1)|1].mx);
    tree[i].sum=tree[i<<1].sum+tree[(i<<1)|1].sum;
}
void dfs2(int u,int len){
    int heavy=0;
    for(node *p=adj[u];p;p=p->next)
        if(p->v!=f[u][0]&&size[heavy]<size[p->v])
            heavy=p->v;
    if(!heavy){
        int tp=u,i;
        for(i=1;i<len;i++)
            tp=f[tp][0];
        root[tp]=NewTree(len);
        length[tp]=len;
        for(i=len;i;i--){
            bl[u]=tp;
            pos[u]=i;
            insert(root[tp],1,1,len,i,wt[u]);
            u=f[u][0];
        }
        return;
    }
    dfs2(heavy,len+1);
    for(node *p=adj[u];p;p=p->next)
        if(p->v!=f[u][0]&&p->v!=heavy)
            dfs2(p->v,1);
}
void init(){
    int i,u,v;
    Read(n);
    for(i=1;i<n;i++){
        Read(u),Read(v);
        addedge(u,v);
        addedge(v,u);
    }
    for(i=1;i<=n;i++)
        Read(wt[i]);
    dfs1(1);
    dfs2(1,1);
}
int LCA(int a,int b){
    if(dep[a]<dep[b])
        swap(a,b);
    int i;
    for(i=MAXLOG;i>=0;i--)
        if(dep[a]-(1<<i)>=dep[b])
            a=f[a][i];
    if(a==b)
        return a;
    for(i=MAXLOG;i>=0;i--)
        if(f[a][i]!=f[b][i])
            a=f[a][i],b=f[b][i];
    return f[a][0];
}
int Get_max(seg_tree *tree,int i,int l,int r,int ll,int rr){
    if(l>rr||r<ll)
        return -INF;
    if(l>=ll&&r<=rr)
        return tree[i].mx;
    int mid=(l+r)>>1;
    int a=Get_max(tree,i<<1,l,mid,ll,rr);
    return max(a,Get_max(tree,(i<<1)|1,mid+1,r,ll,rr));
}
int Query_max(int lca,int a){
    int ret=-INF;
    while(lca!=a){
        if(bl[lca]!=bl[a]){
            ret=max(ret,Get_max(root[bl[a]],1,1,length[bl[a]],1,pos[a]));
            a=f[bl[a]][0];
        }
        else{
            ret=max(ret,Get_max(root[bl[a]],1,1,length[bl[a]],pos[lca]+1,pos[a]));
            break;
        }
    }
    return ret;
}
int Get_sum(seg_tree *tree,int i,int l,int r,int ll,int rr){
    if(l>rr||r<ll)
        return 0;
    if(l>=ll&&r<=rr)
        return tree[i].sum;
    int mid=(l+r)>>1;
    return Get_sum(tree,i<<1,l,mid,ll,rr)+Get_sum(tree,(i<<1)|1,mid+1,r,ll,rr);
}
int Query_sum(int lca,int a){
    int ret=0;
    while(lca!=a){
        if(bl[lca]!=bl[a]){
            ret+=Get_sum(root[bl[a]],1,1,length[bl[a]],1,pos[a]);
            a=f[bl[a]][0];
        }
        else{
            ret+=Get_sum(root[bl[a]],1,1,length[bl[a]],pos[lca]+1,pos[a]);
            break;
        }
    }
    return ret;
}
void solve(){
    Read(m);
    char s[10];
    int lca,a,b;
    while(m--){
        scanf("%s",s);
        Read(a),Read(b);
        if(s[1]=='M'){
            lca=LCA(a,b);
            printf("%d\n",max(wt[lca],max(Query_max(lca,a),Query_max(lca,b))));
        }
        else if(s[1]=='S'){
            lca=LCA(a,b);
            printf("%d\n",Query_sum(lca,a)+Query_sum(lca,b)+wt[lca]);
        }
        else
            insert(root[bl[a]],1,1,length[bl[a]],pos[a],wt[a]=b);
    }
}
int main()
{
    init();
    solve();
}