【线段树合并】BZOJ - 2733 - [HNOI2012]永无乡

题目链接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;
        }
    }
}