HNOI2012永无乡

题目描述
永无乡包含 n 座岛，编号从 1 到 n，每座岛都有自己的独一无二的重要度，按照重要度可 以将这 n 座岛排名，名次用 1 到 n 来表示。某些岛之间由巨大的桥连接，通过桥可以从一个岛 到达另一个岛。如果从岛 a 出发经过若干座（含 0 座）桥可以到达岛 b，则称岛 a 和岛 b 是连 通的。
现在有两种操作：
B x y 表示在岛 x 与岛 y 之间修建一座新桥。
Q x k 表示询问当前与岛 x连通的所有岛中第 k 重要的是哪座岛，即所有与岛 x 连通的岛中重要度排名第 k 小的岛是哪 座，请你输出那个岛的编号。
输入输出格式
输入格式：
输入文件第一行是用空格隔开的两个正整数 n 和 m，分别 表示岛的个数以及一开始存在的桥数。接下来的一行是用空格隔开的 n 个数，依次描述从岛 1 到岛 n 的重要度排名。随后的 m 行每行是用空格隔开的两个正整数 ai 和 bi，表示一开始就存 在一座连接岛 ai 和岛 bi 的桥。后面剩下的部分描述操作，该部分的第一行是一个正整数 q， 表示一共有 q 个操作，接下来的 q 行依次描述每个操作，操作的格式如上所述，以大写字母 Q 或B 开始，后面跟两个不超过 n 的正整数，字母与数字以及两个数字之间用空格隔开。 对于 20%的数据 n<=1000,q<=1000 对于 100%的数据 n<=100000,m<=n，q<=300000
输出格式：
对于每个 Q x k 操作都要依次输出一行，其中包含一个整数，表 示所询问岛屿的编号。如果该岛屿不存在，则输出-1。

最近在看并查集的一些应用,看到了这道题,碰巧zlt之前考了一次线段树合并,学习了下,想练练手,就做了一下.
其实不难想(也不难打),我们用并查集维护图的连通性,然后每次合并的时候用线段树合并一下,找第k大就是权值线段树.好像很简单的样子…….

#include<iostream>#include<cstdio>#include<cstring>#include<cmath>#include<cstdlib>#include<algorithm>using namespace std;typedef long long LL;#define REP(i,a,b) for(register int i = (a), i##_end_ = (b); i <= i##_end_; ++i)#define DREP(i,a,b) for(register int i = (a),i##_end_ = (b); i >= i##_end_; --i)template<typename T> inline bool chkmin(T &a,const T &b) { return a > b ? a = b ,1 : 0;}template<typename T> inline bool chkmax(T &a,const T &b) { return a < b ? a = b, 1 : 0;}int read(){    register int f = 1,s = 0;char c = getchar();    while(!isdigit(c)) { if(c == '-')f = -1; c = getchar();}    while(isdigit(c)) { s = s * 10 + c - '0';c = getchar();}    return f * s;}const int maxn = 1000010;int n,m;int fa[maxn],sum[maxn<<2],rnk[maxn];#define L(x) ch[x][0]#define R(x) ch[x][1]int rt[maxn<<2],ch[maxn<<2][2],tr[maxn<<2],cnt;void insert(int l,int r,int &p,int x){    if(!p)p = ++cnt;    if(l == r){sum[p]++;return ;}    int mid = (l + r) >> 1;    if(x <= mid)insert(l,mid,L(p),x);    else insert(mid+1,r,R(p),x);    sum[p] = sum[L(p)] + sum[R(p)];}int Combine(int l,int r,int x,int y){    if(!x || !y)return x^y;    if(l == r)    {        sum[x] += sum[y];        return x;    }    int mid = (l + r) >> 1;    L(x) = Combine(l,mid,L(x),L(y));    R(x) = Combine(mid+1,r,R(x),R(y));    sum[x] = sum[L(x)] + sum[R(x)];    return x;}int query(int l,int r,int x,int k){    int s = sum[L(x)];    if(l == r)return l;    int mid = (l + r) >> 1;    if(s >= k)return query(l,mid,L(x),k);    else return query(mid+1,r,R(x),k-s);}int find(int x){return x == fa[x] ? x : fa[x] = find(fa[x]);}void merge(int x,int y){    int f1 = find(x),f2 = find(y);    if(f1 == f2)return ;    fa[f1] = f2;    rt[f2] = Combine(1,n,rt[f2],rt[f1]);}char s[10];int id[maxn];int main(){    //freopen("1.in","r",stdin);    //freopen("1.out","w",stdout);    n = read(),m = read();    REP(i,1,n)        rnk[i] = read(),fa[i] = i;    REP(i,1,m)    {        int x = read(),y = read();        merge(x,y);    }    REP(i,1,n)    {        insert(1,n,rt[find(i)],rnk[i]);        id[rnk[i]] = i;    }    int Q = read();    REP(I,1,Q)    {        cin>>s;        int a = read(),b = read();        if(s[0] == 'Q')        {            int f1 = find(a);            if(sum[rt[f1]] < b)puts("-1");            else            {                int t = query(1,n,rt[f1],b);                cout<<id[t]<<endl;            }        }        else merge(a,b);        }    return 0;}