题目链接https://www.lydsy.com/JudgeOnline/problem.php?id=2733&tdsourcetag=s_pctim_aiomsg
线段树合并模板题
学习一下
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef double db;
const int N=1e5+7;
int t[N*40],ls[N*40],rs[N*40];
int rk[N],fa[N],rt[N],id[N],tot;
char s[5];
int fd(int x){return x==fa[x]?x:fa[x]=fd(fa[x]);}
void ist(int &o,int l,int r,int v){
if(!o) o=++tot;
if(l==r){t[o]++;return;}
int m=l+r>>1;
if(v<=m) ist(ls[o],l,m,v);
if(m<v) ist(rs[o],m+1,r,v);
t[o]=t[ls[o]]+t[rs[o]];
}
int meg(int o,int oo){
if(!o) return oo;
if(!oo) return o;
t[o]+=t[oo];
ls[o]=meg(ls[o],ls[oo]);
rs[o]=meg(rs[o],rs[oo]);
return o;
}
int que(int o,int l,int r,int k){
if(t[o]<k) return -1;
if(l==r) return l;
int m=l+r>>1;
if(t[ls[o]]>=k) return que(ls[o],l,m,k);
else return que(rs[o],m+1,r,k-t[ls[o]]);
}
int main(){
int n,m,q,u,v;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&rk[i]);
id[rk[i]]=i;
fa[i]=i;
ist(rt[i],1,n,rk[i]);
}
while(m--){
scanf("%d%d",&u,&v);
u=fd(u),v=fd(v);
if(u==v) continue;
rt[u]=meg(rt[u],rt[v]);
fa[v]=u;
}
scanf("%d",&q);
while(q--){
scanf("%s",s);
if(s[0]=='Q'){
scanf("%d%d",&u,&v);
u=fd(u);
int ans=que(rt[u],1,n,v);
if(ans!=-1) ans=id[ans];
printf("%d\n",ans);
}
else{
scanf("%d%d",&u,&v);
u=fd(u),v=fd(v);
rt[u]=meg(rt[u],rt[v]);
fa[v]=u;
}
}
}