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;
}