[SDOI2014]旅行

看到大家的题解，没看懂\(qwq\)，所以我来发一下题解

首先，线段树要动态开点，不然对于每个宗教开一个线段树会炸的。这个树剖是肯定要敲的。对于每次记录一下每一个宗教的\(Root\)，运用主席树的思想维护

最后就是前排\(orz\) @我是一个菜鸡 是他帮我调出来了几个十分搞笑的错误，帮他安利一下他的题解（应该就在楼下）

具体看代码实现吧，\(40min160\)多行，不过写的还是挺清楚的，总体时间复杂度\(O(nlogn)\)，空间复杂度\(O(nlogn)\)

\(Code\ Below:\)

#include <bits/stdc++.h>
using namespace std;
const int maxn=100000+10;
int n,q,w[maxn],c[maxn],T[maxn],sz;
int head[maxn],to[maxn<<1],nxt[maxn<<1],tot;
int dep[maxn],top[maxn],son[maxn],siz[maxn],fa[maxn],id[maxn],mp[maxn],tim;
struct node{
    int lson,rson,val,sum;
    #define mid (l+r>>1)
    #define lson(rt) tree[rt].lson
    #define rson(rt) tree[rt].rson
    #define val(rt) tree[rt].val
    #define sum(rt) tree[rt].sum
}tree[maxn*24];

inline int read(){
    register int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return (f==1)?x:-x;
}
inline void add(int x,int y){//连边
    to[++tot]=y;
    nxt[tot]=head[x];
    head[x]=tot;
}

void dfs1(int x,int f){//处理深度、重儿子、子树大小和父亲
    dep[x]=dep[f]+1;
    siz[x]=1;
    fa[x]=f;
    int maxson=-1;
    for(int i=head[x],y;i;i=nxt[i]){
        y=to[i];
        if(y==f) continue;
        dfs1(y,x);
        siz[x]+=siz[y];
        if(maxson<siz[y]){
            maxson=siz[y];
            son[x]=y;
        }
    }
}

void dfs2(int x,int topf){//处理重链轻边
    id[x]=++tim;
    mp[tim]=x;
    top[x]=topf;
    if(son[x]) dfs2(son[x],topf);
    for(int i=head[x],y;i;i=nxt[i]){
        y=to[i];
        if(y==fa[x]||y==son[x]) continue;
        dfs2(y,y);//不能写dfs2(y,x)!
    }
}

void pushup(int rt){//上传操作
    val(rt)=max(val(lson(rt)),val(rson(rt)));
    sum(rt)=sum(lson(rt))+sum(rson(rt));
}
void pushdown(int rt){//删除结点
    val(rt)=sum(rt)=0;
}
void update(int &rt,int x,int l,int r,int v){//动态开点
    if(!rt) rt=++sz;
    if(l == r){
        val(rt)=sum(rt)=v;
        return ;
    }
    if(x <= mid) update(lson(rt),x,l,mid,v);
    else update(rson(rt),x,mid+1,r,v);
    pushup(rt);
}
void del(int &rt,int x,int l,int r){//删除结点
	if(!rt) return ;//剪枝
    if(l == r){
        pushdown(rt);
        return ;
    }
    if(x <= mid) del(lson(rt),x,l,mid);
    else del(rson(rt),x,mid+1,r);
    pushup(rt);
}
int query_sum(int &rt,int L,int R,int l,int r){//查询区间和
    if(!rt) return 0;
    if(L <= l && r <= R){
        return sum(rt);
    }
    int ans=0;
    if(L <= mid) ans+=query_sum(lson(rt),L,R,l,mid);
    if(R > mid) ans+=query_sum(rson(rt),L,R,mid+1,r);//不能写else! 
    return ans;
}
int query_max(int &rt,int L,int R,int l,int r){//查询区间最大值
    if(!rt) return 0;
    if(L <= l && r <= R){
        return val(rt);
    }
    int ans=0;
    if(L <= mid) ans=max(ans,query_max(lson(rt),L,R,l,mid));
    if(R > mid) ans=max(ans,query_max(rson(rt),L,R,mid+1,r));//不能写else! 
    return ans;
}

int main()
{
    n=read(),q=read();int x,y,k,ans;
    for(int i=1;i<=n;i++)
        w[i]=read(),c[i]=read();
    for(int i=1;i<n;i++){
        x=read(),y=read();
        add(x,y);add(y,x);
    }
    dfs1(1,0);dfs2(1,1);
    for(int i=1;i<=n;i++)//预处理
        update(T[c[mp[i]]],i,1,n,w[mp[i]]);
    char opt[50];
    while(q--){
        scanf("%s",opt);
        if(opt[1]=='C'){//更改宗教
            x=read(),k=read();
            del(T[c[x]],id[x],1,n);
            c[x]=k;
            update(T[c[x]],id[x],1,n,w[x]);
        }
        if(opt[1]=='W'){//更改评分
            x=read(),k=read();
            del(T[c[x]],id[x],1,n);
            w[x]=k;
            update(T[c[x]],id[x],1,n,w[x]);
        }
        if(opt[1]=='S'){//树剖求和
            x=read(),y=read();
            k=c[x];ans=0;
            while(top[x]!=top[y]){
                if(dep[top[x]]<dep[top[y]]) swap(x,y);//不能写dep[x]<dep[y]!
                ans+=query_sum(T[k],id[top[x]],id[x],1,n);
                x=fa[top[x]];
            }
            if(dep[x]>dep[y]) swap(x,y);
            ans+=query_sum(T[k],id[x],id[y],1,n);
            printf("%d\n",ans);
        }
        if(opt[1]=='M'){//树剖求最大值
            x=read(),y=read();
            k=c[x];ans=0;
            while(top[x]!=top[y]){
                if(dep[top[x]]<dep[top[y]]) swap(x,y);//不能写dep[x]<dep[y]!
                ans=max(ans,query_max(T[k],id[top[x]],id[x],1,n));
                x=fa[top[x]];
            }
            if(dep[x]>dep[y]) swap(x,y);
            ans=max(ans,query_max(T[k],id[x],id[y],1,n));
            printf("%d\n",ans);
        }
    }
    return 0;
}