[bzoj3123][洛谷P3302] [SDOI2013]森林（树上主席树+启发式合并）

传送门

 

突然发现好像没有那么难……https://blog.csdn.net/stone41123/article/details/78167288

首先有两个操作，一个查询，一个连接

查询的话，直接在树上建主席树

然后难点在于连接

用启发式合并就可以了（想了半天都没想出来）

每次合并时，我们把小的树接到大的上，然后dfs一遍小的树，更新信息

然后注意数组……别太小也别太大……（被数组大小坑了好几次提交）

//minamoto
#include<bits/stdc++.h>
using namespace std;
#define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
char buf[1<<21],*p1=buf,*p2=buf;
inline int read(){
    #define num ch-'0'
    char ch;bool flag=0;int res;
    while(!isdigit(ch=getc()))
    (ch=='-')&&(flag=true);
    for(res=num;isdigit(ch=getc());res=res*10+num);
    (flag)&&(res=-res);
    #undef num
    return res;
}
char obuf[1<<24],*o=obuf;
void print(int x){
    if(x>9) print(x/10);
    *o++=x%10+48;
}
const int N=80005,M=N*200;
int ver[N<<2],Next[N<<2],head[N];
int a[N],fa[N],sz[N],b[N];
int n,m,tot,q,size,ans;
void add(int u,int v){
    ver[++tot]=v,Next[tot]=head[u],head[u]=tot;
    ver[++tot]=u,Next[tot]=head[v],head[v]=tot;
}
int L[M],R[M],sum[M],rt[N],cnt;
void update(int last,int &now,int l,int r,int x){
    sum[now=++cnt]=sum[last]+1;
    if(l==r) return;
    int mid=(l+r)>>1;
    if(x<=mid) R[now]=R[last],update(L[last],L[now],l,mid,x);
    else L[now]=L[last],update(R[last],R[now],mid+1,r,x);
}
int query(int u,int v,int lca,int lca_fa,int l,int r,int k){
    if(l>=r) return l;
    int x=sum[L[v]]+sum[L[u]]-sum[L[lca]]-sum[L[lca_fa]];
    int mid=(l+r)>>1;
    if(x>=k) return query(L[u],L[v],L[lca],L[lca_fa],l,mid,k);
    else return query(R[u],R[v],R[lca],R[lca_fa],mid+1,r,k-x);
}
inline int hash(int x){
    return lower_bound(b+1,b+1+size,x)-b;
}
int ff(int x){
    return fa[x]==x?x:fa[x]=ff(fa[x]);
}
int st[N][17],d[N],vis[N];
void dfs(int u,int father,int root){
    st[u][0]=father;
    for(int i=1;i<=16;++i)
    st[u][i]=st[st[u][i-1]][i-1];
    ++sz[root];
    d[u]=d[father]+1;
    fa[u]=root;
    vis[u]=1;
    update(rt[father],rt[u],1,size,hash(a[u]));
    for(int i=head[u];i;i=Next[i]){
        int v=ver[i];
        if(v==father) continue;
        dfs(v,u,root);
    }
}
int LCA(int x,int y){
    if(x==y) return x;
    if(d[x]<d[y]) swap(x,y);
    for(int i=16;i>=0;--i){
        if(d[st[x][i]]>=d[y]) x=st[x][i];
    }
    if(x==y) return x;
    for(int i=16;i>=0;--i){
        if(st[x][i]!=st[y][i])
        x=st[x][i],y=st[y][i];
    }
    return st[x][0];
}
int main(){
    //freopen("testdata.in","r",stdin);
    int t=read();
    n=read(),m=read(),q=read();
    for(int i=1;i<=n;++i)
    a[i]=b[i]=read(),fa[i]=i;
    sort(b+1,b+1+n);
    size=unique(b+1,b+1+n)-b-1;
    for(int i=1;i<=m;++i){
        int u=read(),v=read();
        add(u,v);
    }
    for(int i=1;i<=n;++i)
    if(!vis[i]) dfs(i,0,i);
    while(q--){
        char ch;int x,y;
        while(!isupper(ch=getc()));
        x=read()^ans,y=read()^ans;
        if(ch=='Q'){
            int k=read()^ans;
            int lca=LCA(x,y);
            ans=b[query(rt[x],rt[y],rt[lca],rt[st[lca][0]],1,size,k)];
            print(ans),*o++='\n';
        }
        else{
            add(x,y);
            int u=ff(x),v=ff(y);
            if(sz[u]<sz[v]) swap(x,y),swap(u,v);
            dfs(y,x,u);
        }
    }
    fwrite(obuf,o-obuf,1,stdout);
    return 0;
}