「BZOJ2733」「洛谷3224」「HNOI2012」永无乡【线段树合并】

题目链接

【洛谷】

题解

很明显是要用线段树合并的。
对于当前的每一个连通块都建立一个权值线段树。
权值线段树处理操作中的$k$大的问题。
如果需要合并，那么就线段树暴力合并，时间复杂度是$nlogn$，均摊下来就是$logn$。
判断联通性的问题就用并查集来解决。
如果在同一个联通块里，就不能合并，否则会出一点问题。

代码

#include <bits/stdc++.h>

using namespace std;

const int N = 3000000 + 6;

int rt[N], id[N], fa[N];
int n, m, v[N], k;
char opt[5];

namespace seg {
int ncnt = 0; 
struct node {
	int lc, rc, sz;
	node() { lc = rc = sz = 0; }
} tr[N << 1];
void pushup(int nod) {
	tr[nod].sz = tr[tr[nod].lc].sz + tr[tr[nod].rc].sz;
}
void ins(int &nod, int l, int r, int k) {
	if (!nod) nod = ++ ncnt; 
	if (l == r) {
		tr[nod].sz = 1; 
		return;
	}
	int mid = (l + r) >> 1;
	if (k <= mid) ins(tr[nod].lc, l, mid, k);
	else ins(tr[nod].rc, mid + 1, r, k);
	pushup(nod);
}
int kth(int x, int l, int r, int rk) {
	if (l == r) return l;
	int mid = (l + r) >> 1;
	if (tr[tr[x].lc].sz >= rk) return kth(tr[x].lc, l, mid, rk);
	else return kth(tr[x].rc, mid + 1, r, rk - tr[tr[x].lc].sz); 
}
int merge(int x, int y) {
	if (!x) return y;
	if (!y) return x;
	tr[x].lc = merge(tr[x].lc, tr[y].lc); 
	tr[x].rc = merge(tr[x].rc, tr[y].rc); 
	pushup(x); 
	return x;
} }

int gf(int x) {
	return x == fa[x] ? x : fa[x] = gf(fa[x]);
}
int main() {
	scanf("%d%d", &n, &m);
	for (int i = 1; i <= n; i ++) {
		scanf("%d", &v[i]);
		fa[i] = i;
		id[v[i]] = i; 
	}
	for (int i = 1, a, b; i <= m; i ++) {
		scanf("%d%d", &a, &b);
		int x = gf(a), y = gf(b);
		fa[x] = y;
	}
	for (int i = 1; i <= n; i ++) seg::ins(rt[gf(i)], 1, n, v[i]);
	scanf("%d", &k);
	while (k --) {
		scanf("%s", opt); int x, y; scanf("%d%d", &x, &y);
		if (opt[0] == 'Q') {
			int z = gf(x);
			if (seg::tr[rt[z]].sz < y) puts("-1");
			else {
				int t = seg::kth(rt[z], 1, n, y);
				printf("%d\n", id[t]);
			}
		} else {
			int a = gf(x), b = gf(y);
			if (a == b) continue; 
			fa[a] = b;
			rt[b] = seg::merge(rt[a], rt[b]);
		}
	}
	return 0;
}