bzoj 1036 树链剖分

//和上一题稍微不同，这题权值在点上。
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int MAXN=30000;
int fa[MAXN+9],son[MAXN+9],lev[MAXN+9],top[MAXN+9],id[MAXN+9],head[MAXN+9],size[MAXN+9],val[MAXN+9];
int max_[MAXN*4+9],sum_[MAXN*4+9],tot,cnt;
struct Edge
{
    int u,v,next;
}edge[MAXN*2+9];
void init()
{
    tot=cnt=0;
    for(int i=0;i<=MAXN;i++) fa[i]=top[i]=i;
    memset(sum_,0,sizeof(sum_));
    memset(head,-1,sizeof(head));
}
void add(int x,int y)
{
    edge[tot].u=x;edge[tot].v=y;
    edge[tot].next=head[x];
    head[x]=tot++;
    edge[tot].u=y;edge[tot].v=x;
    edge[tot].next=head[y];
    head[y]=tot++;
}
void dfs1(int x,int d)
{
    lev[x]=d;
    son[x]=0;
    size[x]=1;
    for(int i=head[x];i!=-1;i=edge[i].next){
        int y=edge[i].v;
        if(y==fa[x]) continue;
        fa[y]=x;
        dfs1(y,d+1);
        size[x]+=size[y];
        if(size[son[x]]<size[y]) son[x]=y;
    }
}
void dfs2(int x,int tp)
{
    top[x]=tp;
    id[x]=++cnt;
    if(son[x]) dfs2(son[x],tp);
    for(int i=head[x];i!=-1;i=edge[i].next){
        int y=edge[i].v;
        if(y==fa[x]||y==son[x]) continue;
        dfs2(y,y);
    }
}
void pushup(int rt)
{
    max_[rt]=max(max_[rt<<1],max_[rt<<1|1]);
    sum_[rt]=sum_[rt<<1]+sum_[rt<<1|1];
}
void build(int l,int r,int rt)
{
    if(l==r){
        max_[rt]=sum_[rt]=val[l];
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    pushup(rt);
}
void update(int id,int c,int l,int r,int rt)
{
    if(l==r){
        max_[rt]=sum_[rt]=c;
        return;
    }
    int mid=(l+r)>>1;
    if(id<=mid) update(id,c,l,mid,rt<<1);
    else update(id,c,mid+1,r,rt<<1|1);
    pushup(rt);
}
int query1(int ql,int qr,int l,int r,int rt)
{
    if(ql<=l&&qr>=r) return max_[rt];
    int mid=(l+r)>>1,ans=-MAXN-9;
    if(ql<=mid) ans=max(ans,query1(ql,qr,l,mid,rt<<1));
    if(qr>mid) ans=max(ans,query1(ql,qr,mid+1,r,rt<<1|1));
    return ans;
}
int query2(int ql,int qr,int l,int r,int rt)
{
    if(ql<=l&&qr>=r) return sum_[rt];
    int mid=(l+r)>>1,ans=0;
    if(ql<=mid) ans+=query2(ql,qr,l,mid,rt<<1);
    if(qr>mid) ans+=query2(ql,qr,mid+1,r,rt<<1|1);
    return ans;
}
int solve(int l,int r,bool f)
{
    int ltp=top[l],rtp=top[r],ans=(f?0:-MAXN-9);
    while(ltp!=rtp){
        if(lev[rtp]<lev[ltp]){
            swap(ltp,rtp);
            swap(l,r);
        }
        if(!f) ans=max(ans,query1(id[rtp],id[r],1,cnt,1));
        else ans+=query2(id[rtp],id[r],1,cnt,1);
        r=fa[rtp];
        rtp=top[r];
    }
    if(lev[r]>lev[l]) swap(l,r);
    if(!f) ans=max(ans,query1(id[r],id[l],1,cnt,1));
    else ans+=query2(id[r],id[l],1,cnt,1);
    return ans;
}
int main()
{
    //freopen("in.txt","r",stdin);
    int n,q;
    scanf("%d",&n);
    init();
    for(int i=1;i<n;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
    }
    dfs1(1,1);
    dfs2(1,1);
    for(int i=1;i<=n;i++){
        int x;
        scanf("%d",&x);
        val[id[i]]=x;
    }
    build(1,cnt,1);
    scanf("%d",&q);
    char ch[10];
    while(q--){
        int x,y;
        scanf("%s%d%d",ch,&x,&y);
        if(ch[0]==‘C‘) update(id[x],y,1,cnt,1);
        else if(ch[1]==‘M‘) printf("%d\n",solve(x,y,0));
        else printf("%d\n",solve(x,y,1));
    }
    return 0;
}