洛谷 P3224 [HNOI2012]永无乡

（splay_tree_tag没有比rb_tree_tag快）

#include<cstdio>
#include<algorithm>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
typedef std::pair<int,int> P;
typedef tree<P,null_type,less<P>,rb_tree_tag,tree_order_statistics_node_update> S;
S dat[100100];
int fa[100100],n,m;
char s[10];
int find(int x)
{
    return fa[x]==x ? x : fa[x]=find(fa[x]);
}
int main()
{
    int i,t,ta,tb,a,b,q;
    S::iterator it;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)    fa[i]=i;
    for(i=1;i<=n;i++)
    {
        scanf("%d",&t);
        dat[i].insert(P(t,i));
    }
    for(i=1;i<=m;i++)
    {
        scanf("%d%d",&a,&b);
        ta=find(a);tb=find(b);
        if(ta==tb)    continue;
        fa[ta]=tb;
        if(dat[ta].size()>dat[tb].size())    swap(dat[ta],dat[tb]);
        for(it=dat[ta].begin();it!=dat[ta].end();++it)    dat[tb].insert(*it);
    }
    scanf("%d",&q);
    while(q--)
    {
        scanf("%s%d%d",s,&a,&b);
        if(s[0]=='Q')
        {
            ta=find(a);
            if(dat[ta].size()<b)    printf("-1
");
            else    printf("%d
",dat[ta].find_by_order(b-1)->second);
        }
        else if(s[0]=='B')
        {
            ta=find(a);tb=find(b);
            if(ta==tb)    continue;
            fa[ta]=tb;
            if(dat[ta].size()>dat[tb].size())    swap(dat[ta],dat[tb]);
            for(it=dat[ta].begin();it!=dat[ta].end();++it)    dat[tb].insert(*it);
        }
    }
    return 0;
}