P3313 [SDOI2014]旅行

P3313 [SDOI2014]旅行

树链剖分+动态线段树（并不是lct）

显然的，我们对于每一个宗教都要维护一个线段树。

（那么空间不是爆炸了吗）

在这里引入：动态开点线段树

就是需要的点开起来，不需要的就不开。

其他地方和正常的线段树差不多

这样可以省下一堆空间。

code:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cctype>
#include<queue>
#define re register
using namespace std;
template <typename T> inline T min(T &a,T &b) {return a<b ?a:b;}
template <typename T> inline T max(T &a,T &b) {return a>b ?a:b;}
template <typename T> inline void read(T &x){
    char c=getchar(); x=; bool f=;
    while(!isdigit(c)) f= !f||c=='-' ? :,c=getchar();
    while(isdigit(c)) x=(x<<)+(x<<)+(c^),c=getchar();
    x= f ? x:-x;
}
template <typename T> inline void output(T x){
    if(!x) {putchar(); return ;}
    if(x<) putchar('-'),x=-x;
    int wt[],l=;
    while(x) wt[++l]=x%,x/=;
    while(l) putchar(wt[l--]+);
}
typedef int arr[];
struct data{ //结构体存点
    int l,r,mxd,sum;
    void clear() {l=r=mxd=sum=;}
}a[]; //尽量开大
queue <int> lit;
int n,m,cnt,tot,u;
arr d,fa,siz,bgs,tp,val,sp,id,hd,ed,rt;
int nxt[],poi[];
inline void add_(int x,int y){
    nxt[ed[x]]=++cnt; hd[x]= hd[x] ? hd[x]:cnt;
    ed[x]=cnt; poi[cnt]=y;
}
inline void modify(int &o,int l,int r,int x,int v){ //引用地址便于修改
    if(!o){
        if(!lit.empty()) o=lit.front(),lit.pop(); //内存池节省空间
        else o=++u;
    }
    if(l==r) {a[o].sum=a[o].mxd=v; return;}
    int mid=l+((r-l)>>);
    if(x<=mid) modify(a[o].l,l,mid,x,v);
    else modify(a[o].r,mid+,r,x,v);
    a[o].sum=a[a[o].l].sum+a[a[o].r].sum;
    a[o].mxd=max(a[a[o].l].mxd,a[a[o].r].mxd);
}
inline void remov(int &o,int l,int r,int x){
    if(l==r) {a[o].clear(); lit.push(o),o=; return;}
    int mid=l+((r-l)>>);
    if(x<=mid) remov(a[o].l,l,mid,x);
    else remov(a[o].r,mid+,r,x);
    a[o].sum=a[a[o].l].sum+a[a[o].r].sum;
    a[o].mxd=max(a[a[o].l].mxd,a[a[o].r].mxd);
    if(!a[o].l&&!a[o].r) a[o].clear(),lit.push(o),o=; //左右子树都空->该点为空->删掉
}
inline int query1(int o,int l,int r,int x1,int x2){
    if(!o) return ; //注意空树要跳出
    if(x1<=l&&r<=x2) return a[o].sum;
    int mid=l+((r-l)>>),res=;
    if(x1<=mid) res+=query1(a[o].l,l,mid,x1,x2);
    if(x2>mid) res+=query1(a[o].r,mid+,r,x1,x2);
    return res;
}
inline int query2(int o,int l,int r,int x1,int x2){
    if(!o) return ;
    if(x1<=l&&r<=x2) return a[o].mxd;
    int mid=l+((r-l)>>),res=;
    if(x1<=mid) res=max(res,query2(a[o].l,l,mid,x1,x2));
    if(x2>mid) res=max(res,query2(a[o].r,mid+,r,x1,x2));
    return res;
}
inline void dfs1(int x,int _fa){
    d[x]=d[_fa]+,fa[x]=_fa,siz[x]=;
    for(int i=hd[x];i;i=nxt[i])
        if(poi[i]!=_fa){
            dfs1(poi[i],x);
            siz[x]+=siz[poi[i]];
            if(siz[bgs[x]]<siz[poi[i]]) bgs[x]=poi[i];
        }
}
inline void dfs2(int x,int _top){
    id[x]=++tot,tp[x]=_top;
    if(siz[x]==) return;
    dfs2(bgs[x],_top);
    for(int i=hd[x];i;i=nxt[i])
        if(poi[i]!=fa[x]&&poi[i]!=bgs[x])
            dfs2(poi[i],poi[i]);
}
inline int ask1(int x,int y){
    int col=val[x],res=;
    while(tp[x]!=tp[y]){
        if(d[tp[x]]<d[tp[y]]) swap(x,y);
        res+=query1(rt[col],,n,id[tp[x]],id[x]);
        x=fa[tp[x]];
    }if(d[x]>d[y]) swap(x,y);
    return res+query1(rt[col],,n,id[x],id[y]);
}
inline int ask2(int x,int y){
    int col=val[x],res=;
    while(tp[x]!=tp[y]){
        if(d[tp[x]]<d[tp[y]]) swap(x,y);
        res=max(res,query2(rt[col],,n,id[tp[x]],id[x]));
        x=fa[tp[x]];
    }if(d[x]>d[y]) swap(x,y);
    return max(res,query2(rt[col],,n,id[x],id[y]));
}
//------树剖基础操作--------
inline void change1(int x,int v){
    remov(rt[val[x]],,n,id[x]);
    modify(rt[val[x]=v],,n,id[x],sp[x]);
}//宗教改变：从原来那棵树删掉该点，再加到新树上
inline void change2(int x,int v){
    modify(rt[val[x]],,n,id[x],sp[x]=v);
}//评价改变：直接修改
int main(){
    read(n),read(m); int q1,q2; char opt[];
    for(re int i=;i<=n;++i) read(sp[i]),read(val[i]);
    for(re int i=;i<n;++i) read(q1),read(q2),add_(q1,q2),add_(q2,q1);
    dfs1(,); dfs2(,);
    for(re int i=;i<=n;++i) modify(rt[val[i]],,n,id[i],sp[i]);
    for(re int i=;i<=m;++i){
        scanf("%s",opt); read(q1),read(q2);
        if(opt[]=='C'){
            if(opt[]=='C') change1(q1,q2);
            else change2(q1,q2);
        }else{
            if(opt[]=='S') output(ask1(q1,q2));
            else output(ask2(q1,q2));
            putchar('\n');
        }
    }return ;
}