一道水题,我们用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);
}
}
}