【BZOJ】 [HNOI2012]永无乡-并查集+主席树

传送门：BZOJ2733-永无乡

---

题意

永无乡包含 n 座岛，编号从 1 到 n，每座岛都有自己的独一无二的重要度，按照重要度可 以将这 n 座岛排名，名次用 1 到 n 来表示。某些岛之间由巨大的桥连接，通过桥可以从一个岛 到达另一个岛。如果从岛 a 出发经过若干座（含 0 座）桥可以到达岛 b，则称岛 a 和岛 b 是连 通的。现在有两种操作：B x y 表示在岛 x 与岛 y 之间修建一座新桥。Q x k 表示询问当前与岛 x连通的所有岛中第 k 重要的是哪座岛，即所有与岛 x 连通的岛中重要度排名第 k 小的岛是哪 座，请你输出那个岛的编号。

---

数据范围

对于 100%的数据 n≤100000,m≤n，q≤300000

---

题解

主席树+并查集 暴力合并 实测可A

---

代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
typedef long long ll;

int n,m,f[N],rt[N],ch[N*30][2];
int sum[N*30],T,xx,kk;
char aw[2];
int cnt,tag[N],t[N];

inline int find(int x){return f[x]==x? x:f[x]=find(f[x]);}

inline int read()
{
    char ch=getchar();int x=0,t=1;
    while(ch<'0' || ch>'9' ){if(ch=='-') t=-1;ch=getchar();}
    while(ch>='0' && ch<='9'){x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
    return x*t; 
}
inline void add(int &k,int l,int r,int kk)
{
    if(k==0) k=++cnt;
    if(l==r){
        sum[k]=1;return;
    }
    int mid=(l+r)>>1;
    if(kk<=mid) add(ch[k][0],l,mid,kk);
    else add(ch[k][1],mid+1,r,kk);
    sum[k]=sum[ch[k][0]]+sum[ch[k][1]];
}

inline int get(int k,int l,int r,int kk)
{
    if(l==r) return l;
    int mid=(l+r)>>1,ss=sum[ch[k][0]];
    if(kk<=ss) return get(ch[k][0],l,mid,kk);
    else return get(ch[k][1],mid+1,r,kk-ss);
}

inline int merge(int x,int y)
{
    if(!x || !y)return x+y;
    ch[x][0]=merge(ch[x][0],ch[y][0]);
    ch[x][1]=merge(ch[x][1],ch[y][1]);
    sum[x]=sum[ch[x][0]]+sum[ch[x][1]];
    return x;
}

int main(){
    n=read();m=read();
    for(int i=1;i<=n;i++){
        t[i]=read();
        f[i]=i;
    }
    for(int om,ow,i=1;i<=m;i++){
        om=read();ow=read();
        f[find(om)]=find(ow);
    }
    for(int i=1;i<=n;i++){
        add(rt[find(i)],1,n,t[i]);
        tag[t[i]]=i;
    }
    T=read();
    while(T--){
        scanf("%s",aw); 
        xx=read();kk=read(); 
        if(aw[0]=='Q'){
            if(sum[rt[find(xx)]]<kk){
                printf("-1\n");continue;
            }
            printf("%d\n",tag[get(rt[find(xx)],1,n,kk)]);
        }else{
            int op=find(xx),oq=find(kk);
            if(op==oq) continue;
            f[oq]=op;
            rt[op]=merge(rt[oq],rt[op]);
        }
    }
    return 0; 
}