2018.07.24 bzoj3531: [Sdoi2014]旅行（树链剖分+动态开点）

传送门

树链剖分。

如何维护？

如果颜色少直接每种颜色一颗线段树走人。

但这题颜色数量不大于1e5" role="presentation" style="position: relative;">1e51e5，空间难以承受。

因此我们动态开点（毕竟所有点数加起来只有1e5" role="presentation" style="position: relative;">1e51e5）

其余就是树链剖分的基本操作了。

代码如下：

#include<bits/stdc++.h>
#define N 3000005
#define lc T[p].l
#define rc T[p].r
using namespace std;
int n,m,cntx=0,cnt=0,tot=0,first[N],dep[N],top[N],num[N],fa[N],hson[N],siz[N],rt[N],w[N],c[N],pos=0;
struct Node{int l,r,sum,maxn;}T[N<<2];
struct node{int v,next;}e[N<<1];
inline void add(int u,int v){e[++cnt].v=v,e[cnt].next=first[u],first[u]=cnt;}
inline void dfs1(int p){
    siz[p]=1,hson[p]=0;
    for(int i=first[p];i;i=e[i].next){
        if(e[i].v==fa[p])continue;
        fa[e[i].v]=p,dep[e[i].v]=dep[p]+1,dfs1(e[i].v),siz[p]+=siz[e[i].v];
        if(siz[hson[p]]<siz[e[i].v])hson[p]=e[i].v;
    }
}
inline void dfs2(int p,int tp){
    top[p]=tp,num[p]=++tot;
    if(hson[p])dfs2(hson[p],tp);
    else return;
    for(int i=first[p];i;i=e[i].next)if(e[i].v!=fa[p]&&e[i].v!=hson[p])dfs2(e[i].v,e[i].v);
}
inline void pushup(int p){
    T[p].maxn=T[p].sum=0;
    if(lc)T[p].maxn=T[lc].maxn,T[p].sum=T[lc].sum;
    if(rc)T[p].maxn=max(T[p].maxn,T[rc].maxn),T[p].sum+=T[rc].sum;
}
inline void update(int&p,int l,int r,int v,int k){
    if(!p)p=++pos;
    if(l==r){T[p].sum+=v,T[p].maxn+=v;return;}
    int mid=l+r>>1;
    if(k<=mid)update(lc,l,mid,v,k);
    else update(rc,mid+1,r,v,k);
    pushup(p);
}
inline int query_sum(int p,int ql,int qr,int l,int r){
    if(!p)return 0;
    if(ql<=l&&r<=qr)return T[p].sum;
    int ret=0,mid=l+r>>1;
    if(ql<=mid)ret+=query_sum(lc,ql,qr,l,mid);
    if(mid<qr)ret+=query_sum(rc,ql,qr,mid+1,r);
    return ret;
}
inline int query_max(int p,int ql,int qr,int l,int r){
    if(!p)return 0;
    if(ql<=l&&r<=qr)return T[p].maxn;
    int ret=0,mid=l+r>>1;
    if(ql<=mid)ret=max(ret,query_max(lc,ql,qr,l,mid));
    if(qr>mid)ret=max(ret,query_max(rc,ql,qr,mid+1,r));
    return ret;
}
inline int ask_sum(int x,int y,int col){
    int ans=0;
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        ans+=query_sum(rt[col],num[top[x]],num[x],1,n);
        x=fa[top[x]];
    }
    if(dep[x]<dep[y])swap(x,y);
    return ans+query_sum(rt[col],num[y],num[x],1,n);
}
inline int ask_max(int x,int y,int col){
    int ans=0;
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]])swap(x,y);
        ans=max(ans,query_max(rt[col],num[top[x]],num[x],1,n));
        x=fa[top[x]];
    }
    if(dep[x]<dep[y])swap(x,y);
    return max(ans,query_max(rt[col],num[y],num[x],1,n));
}
inline int read(){
    int ans=0;
    char ch=getchar();
    while(!isdigit(ch))ch=getchar();
    while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
    return ans;
}
int main(){
    n=read(),m=read();
    for(int i=1;i<=n;++i)w[i]=read(),c[i]=read();
    for(int i=1;i<n;++i){
        int u=read(),v=read();
        add(u,v),add(v,u);
    }
    dfs1(1),dfs2(1,1);
    for(int i=1;i<=n;++i)update(rt[c[i]],1,n,w[i],num[i]);
    while(m--){
        char op[3];
        scanf("%s",op);
        int a=read(),b=read();
        switch(op[1]){
            case 'C':{update(rt[c[a]],1,n,-w[a],num[a]),update(rt[c[a]=b],1,n,w[a],num[a]);break;}
            case 'W':{update(rt[c[a]],1,n,b-w[a],num[a]),w[a]=b;break;}
            case 'S':{cout<<ask_sum(a,b,c[a])<<'\n';break;}
            default:{cout<<ask_max(a,b,c[a])<<'\n';break;}
        }
    }
    return 0;
}