【BZOJ 3123】 [Sdoi2013]森林 主席树启发式合并

我们直接按父子关系建主席树，然后记录倍增方便以后求LCA，同时用并查集维护根节点，而且还要记录根节点对应的size，用来对其启发式合并，然后每当我们合并的时候我们都要暴力拆小的一部分重复以上部分，总时间复杂度为O(n*log)，因为每个的节点只会作为小的部分合并，因此他所在的一小部分至少变大2倍，对于每一个节点他作为被合并方最多log次，因此其复杂度为O(n*log)，而这个是绝对跑不满还差很多的，我们视他为无常数就好了，当然这一切都是建立在无拆分操作的基础之上，只要有了拆分启发式合并的复杂度就是暴力了。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define MAXN 80001
using namespace std;
inline int read(){
    register int sum=;register char ch=getchar();
    while(ch<''||ch>'')ch=getchar();
    while(ch>=''&&ch<='')sum=(sum<<)+(sum<<)+ch-'',ch=getchar();
    return sum;
}
int n,m,T,fa[MAXN],size[MAXN],f[MAXN][];
struct Via{
    int to,next;
}c[MAXN<<];
int head[MAXN],t;
struct Seg_Tree{
    Seg_Tree *ch[];
    int size;
}*root[MAXN],*null;
int pos[MAXN],a[MAXN],deep[MAXN];
int Hash[MAXN],len;
bool v[MAXN];
//*********************Define********************//
inline int find(int x){
    return x==fa[x]?x:(fa[x]=find(fa[x]));
}
inline void Unit_to_(int x,int y){
    fa[find(x)]=find(y);
}
//*******************Union_Find_Sets*************//
inline void add(int x,int y){
    c[++t].to=y,c[t].next=head[x],head[x]=t;
}
//********************Maintain_Tree**************//
int comp(const int x,const int y){
    return a[x]<a[y];
}
inline void HASH(){
    sort(pos+,pos+n+,comp);
    for(register int i=;i<=n;i++)
        if(i==||a[pos[i]]!=a[pos[i-]])Hash[++len]=a[pos[i]],a[pos[i]]=len;
        else a[pos[i]]=len;
}
//********************Hash**********************//
void ins(Seg_Tree *&p,Seg_Tree *last,int l,int r,int key){
    p=new Seg_Tree,*p=*last,++p->size;if(l==r)return;
    if(key<=((l+r)>>))ins(p->ch[],last->ch[],l,((l+r)>>),key);
    else ins(p->ch[],last->ch[],((l+r)>>)+,r,key);
}
void Dfs_Build(int x,int Fa){
    deep[x]=deep[Fa]+;v[x]=;
    for(register int i=;i<;i++)f[x][i]=f[f[x][i-]][i-];
    ins(root[x],root[Fa],,len,a[x]);
    for(register int i=head[x];i;i=c[i].next)
        if(c[i].to!=Fa) f[c[i].to][]=x,Dfs_Build(c[i].to,x);
}
int query(Seg_Tree *A1,Seg_Tree *A2,Seg_Tree *B1,Seg_Tree *B2,int l,int r,int k){
    if(l==r)return l;register int sum=A1->ch[]->size-A2->ch[]->size+B1->ch[]->size-B2->ch[]->size;
    if(sum>=k)return query(A1->ch[],A2->ch[],B1->ch[],B2->ch[],l,((l+r)>>),k);
    else return query(A1->ch[],A2->ch[],B1->ch[],B2->ch[],((l+r)>>)+,r,k-sum);
}
void Del(Seg_Tree *p,Seg_Tree *last){
    if(p==last)return;
    Del(p->ch[],last->ch[]),Del(p->ch[],last->ch[]);
    delete p;
}
void Delete(int x,int fa){
    for(register int i=head[x];i;i=c[i].next)
        if(c[i].to!=fa)Delete(c[i].to,x);
    Del(root[x],root[fa]),root[x]=null;
}
//********************Seg_Tree******************//
inline int Lca(int x,int y){
    if(deep[x]<deep[y])x^=y^=x^=y;
    register int k=deep[x]-deep[y];
    for(register int i=;(<<i)<=k;i++)
        if(k&(<<i)) x=f[x][i];
    if(x==y)return x;
    for(register int i=;i>=;i--)
        if(f[x][i]!=f[y][i])
            x=f[x][i],y=f[y][i];
    return f[x][];
}
//*********************LCA*********************//
inline void Init(){
    scanf("%*d"),n=read(),m=read(),T=read();
    null=new Seg_Tree,null->ch[]=null->ch[]=null,null->size=,root[]=null;
    for(register int i=;i<=n;i++)a[i]=read(),pos[i]=i,root[i]=null,size[i]=,fa[i]=i;HASH();
    for(register int i=,x,y;i<=m;i++)x=read(),y=read(),add(x,y),add(y,x),size[find(y)]+=size[find(x)],Unit_to_(x,y);
    for(register int i=;i<=n;i++)if(!v[i])Dfs_Build(find(i),);
}
inline void Work(){
    register int x,y,k,lca,ans=;register char s[];
    while(T--){
        scanf("%s",s);
        if(s[]=='Q'){
            x=read()^ans,y=read()^ans,k=read()^ans,lca=Lca(x,y);
            printf("%d\n",(ans=Hash[query(root[x],root[lca],root[y],root[f[lca][]],,len,k)]));
        }else{
            x=read()^ans,y=read()^ans;if(size[find(x)]>size[find(y)])x^=y^=x^=y;
            Delete(find(x),),size[find(y)]+=size[find(x)],Unit_to_(x,y);
            f[x][]=y,add(x,y),add(y,x),Dfs_Build(x,y);
        }
    }
}
//*********************Main********************//
int main(){
    Init();
    Work();
    return ;
}