BZOJ2733: [HNOI2012]永无乡

n<=1e5个点，每个点有排名，m次连接把某两个岛屿连起来，然后q次操作：合并两个岛屿；查询某个岛屿能到的所有岛屿的第k名次是哪一个。

线段树合并，并查集！

小坑：合并两棵线段树之前，记得判断这两棵树是不是已经在一起了，如果是就不理他；

合并线段树用的是并查集根节点对应的线段树。

#include<cstring>
#include<cstdlib>
#include<cstdio>
//#include<assert.h>
#include<math.h>
#include<algorithm>
//#include<iostream>
using namespace std;

int n,m,q;
#define maxn 100011
struct SMT
{
    struct Node
    {
        int cnt;
        int l,r;
        int ls,rs;
    }a[maxn*20];
    int size;
    SMT() {size=0;}
    void up(int x)
    {
        const int &p=a[x].ls,&q=a[x].rs;
        a[x].cnt=a[p].cnt+a[q].cnt;
    }
    void build(int &x,int L,int R,int pos)
    {
        x=++size;
        a[x].l=L;a[x].r=R;
        if (L==R)
        {
            a[x].cnt=1;
            a[x].ls=a[x].rs=0;
        }
        else
        {
            const int mid=(L+R)>>1;
            if (pos<=mid) build(a[x].ls,L,mid,pos);
            else build(a[x].rs,mid+1,R,pos);
            up(x);
        }
    }
    void build(int &x,int pos) {build(x,1,n,pos);}
    int query(int x,int k)
    {
        if (a[x].l==a[x].r) return a[x].l;
        if (a[a[x].ls].cnt<k) return query(a[x].rs,k-a[a[x].ls].cnt);
        return query(a[x].ls,k);
    }
    int combine(int x,int y)
    {
        if (!x || !y) return x+y;
        a[x].ls=combine(a[x].ls,a[y].ls);
        a[x].rs=combine(a[x].rs,a[y].rs);
//        cout<<a[x].cnt<<','<<a[y].cnt<<":)";
        a[x].cnt+=a[y].cnt;
        return x;
    }
}t;
int root[maxn],rank[maxn],id[maxn];
int ufs[maxn];
int Find(int x) {return x==ufs[x]?x:(ufs[x]=Find(ufs[x]));}
void Union(int x,int y)
{
    x=Find(x),y=Find(y);
    if (x==y) return;
    ufs[x]=y;
}
bool isdigit(char c) {return c>='0' && c<='9';}
int qread()
{
    char c;int s=0;while (!isdigit(c=getchar()));
    do s=s*10+c-'0'; while (isdigit(c=getchar()));return s;
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++) id[(rank[i]=qread())]=ufs[i]=i;
    for (int i=1;i<=n;i++) t.build(root[i],rank[i]);
    for (int i=1,x,y;i<=m;i++)
    {
        x=Find(qread()),y=Find(qread());
        if (x!=y) root[x]=t.combine(root[x],root[y]),Union(y,x);
    }
    q=qread();int x,y;
    while (q--)
    {
        char c=getchar();while (c!='B' && c!='Q') c=getchar();
        if (c=='Q')
        {
            x=Find(qread()),y=qread();
            if (t.a[root[x]].cnt<y) puts("-1");
            else printf("%d
",id[t.query(root[x],y)]);
        }
        else
        {
            x=Find(qread()),y=Find(qread());
            if (x!=y) root[x]=t.combine(root[x],root[y]),Union(y,x);
        }
//        for (int j=1;j<=n;j++) cout<<t.a[root[Find(j)]].cnt<<' ';cout<<endl;
    }
    return 0;
}