P3224 [HNOI2012]永无乡【并查集、线段树合并】

传送门

在线维护连通块里面权值的第$K$大

分析

维护集合第K大，由于每个点的权值都是不同的，且值域范围$1$~$N$，考虑使用动态开点权值线段树维护。
剩下的就是，如何在线维护加入一个集合到一个集合里面的数据，线段树合并能够在此条件下有良好的复杂度，实现将一个集合里面的桶合并到另一个集合里面（权值线段树就是一个桶线段树）。
注意，在并查集维护点联通的时候，线段树合并也要朝着并查集合并的防线进行合并。
比如有操作$father[y]=x$ 此时 $root[x]=merge(root[x],root[y])$即可
以下为代码

代码

//P3224
/*
  @Author: YooQ
*/
#include <bits/stdc++.h>
using namespace std;
#define sc scanf
#define pr printf
#define ll long long
#define FILE_OUT freopen("out", "w", stdout);
#define FILE_IN freopen("in", "r", stdin);
#define debug(x) cout << #x << ": " << x << "\n";
#define AC 0
#define WA 1
#define INF 0x3f3f3f3f
const ll MAX_N = 1e5+5;
const ll MOD = 1e9+7;
int N, M, K;


struct Tr {
	int k, l, r;
}tr[MAX_N*60];
int root[MAX_N];
int indx = 0;

int mk() {
	return ++indx;
}

void push_up(int rt) {
	tr[rt].k = tr[tr[rt].l].k + tr[tr[rt].r].k;
}

void update(int &rt, int l, int r, int x) {
	if (!rt) rt = mk();
	if (l == r) {
		tr[rt].k = 1;
		return;
	}
	int mid = l + ((r-l)>>1);
	if (x <= mid) update(tr[rt].l, l, mid, x);
	if (x  > mid) update(tr[rt].r, mid+1, r, x);
	push_up(rt);
}

int merge(int x, int y, int l, int r) {
	if (!x || !y) return x | y;
	if (l == r) {
		tr[x].k += tr[y].k;
		return x;
	}
	int mid = l + ((r-l)>>1);
	tr[x].l = merge(tr[x].l, tr[y].l, l, mid);
	tr[x].r = merge(tr[x].r, tr[y].r, mid+1, r);
	push_up(x);
	return x;
}

int query(int rt, int l, int r, int x) {
	if (!rt || tr[rt].k < x) return 0;
	if (l == r) {
		return l;
	}
	int lsum = tr[tr[rt].l].k;
	int mid = l + ((r-l)>>1);
	if (lsum >= x) return query(tr[rt].l, l, mid, x);
	else return query(tr[rt].r, mid+1, r, x-lsum);
}

int father[MAX_N];

int find(int x) {
	if (x == father[x]) {
		return x;
	}
	int p = find(father[x]);
//	merge(root[p], root[x], 1, N);
	father[x] = p;
	return father[x];
}

void unit(int x, int y) {
	x = find(x);
	y = find(y);
	if (x != y) {
		father[y] = x;
		root[x] = merge(root[x], root[y], 1, N);
	}
}

int arr[MAX_N];
int brr[MAX_N];

void solve(){
	sc("%d%d", &N, &M);
	for (int i = 1; i <= N; ++i) {
		father[i] = i;
		sc("%d", &arr[i]);
		update(root[i], 1, N, arr[i]);
		brr[arr[i]] = i;
	}
	brr[0] = -1;
	int u, v;
	for (int i = 1; i <= M; ++i) {
		sc("%d%d", &u, &v);
		unit(u, v);
	}
	char opt[10];
	sc("%d", &K);
	for (int i = 1; i <= K; ++i) {
		sc("%s", opt);
		sc("%d%d", &u, &v);
		if (*opt == 'Q') {
			pr("%d\n", brr[query(root[find(u)], 1, N, v)]);
		} else {
			unit(u, v);
		}
	}
}

/*
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

*/
signed main()
{
	#ifndef ONLINE_JUDGE
	//FILE_IN
	FILE_OUT
	#endif
	int T = 1;//cin >> T;
	while (T--) solve();

	return AC;
}