bzoj 2733 splay+啓發式合併

一道水題，我們用splay+啓發式合併（其實就是一個比較暴力的方法，把siz小的splay樹裏的點一個一個塞到大的splay樹中）

最開始我想了半天怎麼存root，後來發現我傻了，直接splay到根就好了。

這份代碼裏德root數組其實是並查集數組，不要在意細節- -。

下面是我AC代碼：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define N 100005
using namespace std;
int n,m,c[N][2],val[N],fa[N],root[N],siz[N],a,b;
void update(int x){
    int l=c[x][0],r=c[x][1];
    siz[x]=siz[l]+siz[r]+1;
}
void rotate(int x){    
    int y=fa[x],z=fa[y];    
    int l=(c[y][0]!=x),r=l^1;    
    if(fa[y]){    
       if(c[z][0]==y) c[z][0]=x;    
       else c[z][1]=x;    
    }    
    fa[x]=z,fa[y]=x,fa[c[x][r]]=y;    
    c[y][l]=c[x][r],c[x][r]=y;    
    update(y),update(x);  
}    
void splay(int x){
    while(fa[x]){
        int y=fa[x],z=fa[y];
        if(fa[y]){
            if(c[y][0]==x^c[z][0]==y)rotate(x);
            else rotate(y); 
        }
        rotate(x);
    }
}
int insert(int x,int pos){
    int l=c[x][0],r=c[x][1];
    if(val[pos]>val[x]){
        if(r)return insert(r,pos);
        else {
            c[pos][0]=c[pos][1]=0,siz[pos]=1,c[x][1]=pos,fa[pos]=x,update(x),splay(pos);
            return pos;
        }
    }
    else{
        if(l)return insert(l,pos);
        else {
            c[pos][0]=c[pos][1]=0,siz[pos]=1,c[x][0]=pos,fa[pos]=x,update(x),splay(pos);
            return pos;
        }
    }
}
void merge(int x,int rt){
    int l=c[x][0],r=c[x][1];
    rt=insert(rt,x);
    if(l) merge(l,rt);
    if(r) merge(r,rt);
}
char s[2];
int query(int x,int rk){
    if(siz[x]<rk)return -1;
    int l=c[x][0],r=c[x][1];
    if(siz[l]+1==rk)return x;
    if(siz[l]>=rk)return query(l,rk);
    return query(r,rk-siz[l]-1);
}
int find(int x){return x==root[x]?x:root[x]=find(root[x]);}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&val[i]);
        siz[i]=1;root[i]=i;
    }
    for(int i=1;i<=m;i++){
        scanf("%d%d",&a,&b);
        int x=find(a),y=find(b);
        if(x==y)continue;
        else{
            root[x]=y;
            splay(a),splay(b);
            if(siz[a]>siz[b])swap(a,b);
            merge(a,b);
        }
    }
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%s%d%d",s,&a,&b);
        if(s[0]=='Q'){
            splay(a);
            printf("%d\n",query(a,b));
        }else{
            int x=find(a),y=find(b);
            if(x==y)continue;
            root[x]=y;
            splay(a),splay(b);
            if(siz[a]>siz[b])swap(a,b);
            merge(a,b);
        }
    }
}