bzoj3531 [Sdoi2014]旅行(树链剖分+动态开点线段树)

我们给每个宗教都建一棵线段树，维护区间和和区间最大值。但是这样会MLE,因此我们掌握了新技能：动态开点线段树。也就是不预先把所有点都建出来，用到才建。每次最多建logn个点，所以开到mlogn就足够了。这样还要记录每棵树的左右儿子，因为他们已不再是p<<1,p<<1|1，而是任意的点了。

#include 
using namespace std;
#define N 100010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,w[N],c[N],root[N],h[N],num=0,cnt=0;
int dep[N],fa[N],son[N],size[N],id[N],dfn=0,top[N];
struct edge{
    int to,next;
}data[N<<1];
inline void add(int x,int y){
    data[++num].to=y;data[num].next=h[x];h[x]=num;
}
struct node{
    int lc,rc,mx,sum;
}tree[2000010];
void dfs1(int x){
    size[x]=1;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;
        if(y==fa[x]) continue;
        fa[y]=x;dep[y]=dep[x]+1;dfs1(y);size[x]+=size[y];
        if(size[y]>size[son[x]]) son[x]=y;
    }
}
void dfs2(int x,int tp){
    id[x]=++dfn;top[x]=tp;
    if(son[x]) dfs2(son[x],tp);
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;
        if(y==fa[x]||y==son[x]) continue;
        dfs2(y,y);
    }
}
inline void pushup(int p){
    tree[p].sum=tree[tree[p].lc].sum+tree[tree[p].rc].sum;
    tree[p].mx=max(tree[tree[p].lc].mx,tree[tree[p].rc].mx);
}
void change(int &p,int l,int r,int x,int val){
    if(!p) p=++cnt;
    if(l==r){tree[p].mx=tree[p].sum=val;return;}
    int mid=l+r>>1;
    if(x<=mid) change(tree[p].lc,l,mid,x,val);
    else change(tree[p].rc,mid+1,r,x,val);
    pushup(p);
}
int qmax(int p,int l,int r,int x,int y){
    if(!p) return 0;
    if(x<=l&&r<=y) return tree[p].mx;
    int mid=l+r>>1,res=0;
    if(x<=mid) res=max(res,qmax(tree[p].lc,l,mid,x,y));
    if(y>mid) res=max(res,qmax(tree[p].rc,mid+1,r,x,y));
    return res;
}
int qsum(int p,int l,int r,int x,int y){
    if(!p) return 0;
    if(x<=l&&r<=y) return tree[p].sum;
    int mid=l+r>>1,res=0;
    if(x<=mid) res+=qsum(tree[p].lc,l,mid,x,y);
    if(y>mid) res+=qsum(tree[p].rc,mid+1,r,x,y);
    return res;
}
int domax(int x,int y,int rt){
    int res=0;
    while(top[x]!=top[y]){
        if(dep[top[x]]1,n,id[top[x]],id[x]));
        x=fa[top[x]];
    }
    if(id[x]>id[y]) swap(x,y);
    return max(res,qmax(rt,1,n,id[x],id[y]));
}
int dosum(int x,int y,int rt){
    int res=0;
    while(top[x]!=top[y]){
        if(dep[top[x]]1,n,id[top[x]],id[x]);
        x=fa[top[x]];
    }
    if(id[x]>id[y]) swap(x,y);
    return res+qsum(rt,1,n,id[x],id[y]);
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();
    for(int i=1;i<=n;++i) w[i]=read(),c[i]=read();
    for(int i=1;iint x=read(),y=read();add(x,y);add(y,x);
    }dep[1]=1;dfs1(1);dfs2(1,1);
    for(int i=1;i<=n;++i) change(root[c[i]],1,n,id[i],w[i]);
    while(m--){
        char op[4];scanf("%s",op+1);int x=read(),y=read();
        if(op[2]=='C') change(root[c[x]],1,n,id[x],0),change(root[c[x]=y],1,n,id[x],w[x]);
        if(op[2]=='W') change(root[c[x]],1,n,id[x],w[x]=y);
        if(op[2]=='S') printf("%d\n",dosum(x,y,root[c[x]]));
        if(op[2]=='M') printf("%d\n",domax(x,y,root[c[x]]));
    }
    return 0;
}