POJ 1988(带权并查集)

POJ 1988

• 题目问的是包括x方块的堆 下面的 所有方块的个数,而不是方块x下面所有方块的个数,
• 将最下面的方块作为根节点（父亲节点设置时带方向）
• 一个数组记录此堆的数量，另一个数组记该点下面方块的个数
• find(x)时，更新所有点的下面方块个数,
• 询问点x的结果时，首先要find(x)，使的结果为该堆的根节点下面的方块数量

#include <iostream>
using namespace std;
const int maxn = 3e4+5;
int tot[maxn],pre[maxn],down[maxn];
void init() {
	for (int i=1; i<maxn; i++) {
		tot[i] = 1; //每棵树大小默认为1 
		pre[i] = i; // 每个点的顶点默认为自己 
		down[i] = 0; // 点i下面有多少个物品
	}
}
int find(int x) {
	if (x != pre[x]) {
		int lastx = pre[x];
		pre[x] = find(pre[x]);
		down[x] += down[lastx]; //两个堆合并时更能体现这个的作用,一个堆find()时，其实值没有改变
	} 
	return pre[x];
}
void merge(int a,int b) {
	int root_a = find(a);
	int root_b = find(b);
	if (root_a != root_b) {
		pre[root_a] = root_b;
		// 每次移动到另一个上面时,只需要更新该堆得最下面一个点的down[]值;通过find()累计所有点的down[] 
		down[root_a] = tot[root_b]; // root_a点下面压的 正好是root_b这个数的所有节点数
		tot[root_b] += tot[root_a];
	}
}
int main() {
	freopen("a.txt","r",stdin);
	int p;
	cin >> p;

	init();

	char ch;
	int u,v;
	for (int k=0; k<p; k++) {
		cin >> ch;
		if (ch == 'M') {
			cin >> u >> v;
			merge(u,v);		
		}
		else {
			cin >> u;
//			printf("询问点%d属于的堆下面有多少个方块\n",u);
			find(u); // 疏通一颗数中所有点的down[]; 
			printf("%d\n",down[u]);
		}
	}
	return 0;
}