bzoj2733 [HNOI2012]永无乡

[HNOI2012]永无乡

Time Limit: 10 Sec Memory Limit: 128 MB

Description

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

Input

输入文件第一行是用空格隔开的两个正整数 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

Output

对于每个 Q x k 操作都要依次输出一行，其中包含一个整数，表 示所询问岛屿的编号。如果该岛屿不存在，则输出-1。

Sample Input

5 1
4 3 2 5 1
1 2
7
Q 3 2
Q 2 1
B 2 3
B 1 5
Q 2 1
Q 2 4
Q 2 3

Sample Output

-1
2
5
1
2

最讨厌数据结构了。。。。
又想起了线段树合并，写一道来回忆一下。。。
就这样吧。。。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 5, INF = 0x7fffffff, L = 0, R = 1;
struct lpl{
	int l, r, data, son[2];
}node[5000005];
int n, m, k, q, aaa, bbb, mx = -INF, mn = INF, cnt = maxn;
int ini[maxn], fa[maxn], root[5 * maxn];
char que[5];

inline int find(int t)
{
	if(fa[t] == t) return fa[t];
	else return fa[t] = find(fa[t]);
}

void insert(int &t, int l, int r, int x)
{
	if(l == r) t = k;
	else t = ++cnt;
	node[t].l = l, node[t].r = r; node[t].data++;
	if(l == r) return;
	int mid = (l + r) / 2;
	if(x <= mid) insert(node[t].son[L], l, mid, x);
	if(x > mid) insert(node[t].son[R], mid + 1, r, x);
}

inline void putit()
{
	scanf("%d%d", &n, &m);
	for(k = 1; k <= n; ++k) scanf("%d" ,&ini[k]), mx = max(mx, ini[k]), mn = min(mn, ini[k]);
	for(int i = 1; i <= n; ++i) fa[i] = i;

}

int connect(int a, int b)
{
	if(a * b == 0) return a + b;
	node[a].data += node[b].data;
	node[a].son[L] = connect(node[a].son[L], node[b].son[L]);
	node[a].son[R] = connect(node[a].son[R], node[b].son[R]);
	return a;
}

int query(int t, int k)
{
	if(node[t].l == node[t].r) return t;
	if(k > node[node[t].son[L]].data) return query(node[t].son[R], k - node[node[t].son[L]].data);
	return query(node[t].son[L], k);
}

inline void workk()
{
	for(k = 1; k <= n; ++k) insert(root[k], mn, mx, ini[k]);
	for(int i = 1; i <= m; ++i) {
		scanf("%d%d", &aaa, &bbb);
		aaa = find(aaa), bbb = find(bbb);
		if(aaa != bbb){
			fa[bbb] = aaa; root[aaa] = connect(root[aaa], root[bbb]);
		}
	}
	scanf("%d", &q);
	for(int i = 1; i <= q; ++i) {
		scanf("%s", que + 1);
		scanf("%d%d", &aaa, &bbb);
		aaa = find(aaa);
		if(que[1] == 'Q'){
			if(node[root[aaa]].data < bbb) printf("-1\n");
			else printf("%d\n", query(root[aaa], bbb));
		}
		if(que[1] == 'B'){
			bbb = find(bbb); 
			if(aaa != bbb){
				fa[bbb] = aaa; root[aaa] = connect(root[aaa], root[bbb]);				
			}
		}
	} 
}

int main()
{
	putit();
	workk();
	return 0;
}