POJ 1988 Cube Stacking(并查集)

题意：

有N堆方块（初始时每堆有一个方块），编号1~N，有两种操作：

M x y:将x所在堆的方块拿起来放到y所在堆的上边。

C x:标号为x的方块下面有多少个方块。

解题思路：暴力肯定会超时，这是一道并查集的题目。每一个方块有三个参数:

par：父亲结点的编号

under：当前结点下面的方块数目，在堆合并和路径压缩中都要更新

sum：如果该结点作为根结点，则表示的是该堆的方块总数

样例演示：

代码：

#include <cstdio>
#include <cstring>
#define M 50000

int par[M], under[M], sum[M];

int get_par(int u) {
if(par[u] != u) {
int tmp = par[u];
par[u] = get_par(par[u]); //更改父亲结点
under[u] += under[tmp]; //更新under参数
}
return par[u];
}

int query(int a, int b) {
return get_par(a) == get_par(b);
}

void merge(int a, int b) {
int t1 = get_par(a);
int t2 = get_par(b);
under[t1] = sum[t2];
sum[t2] += sum[t1];
par[t1] = t2;
}

int main() {
int p;
char cmd[10];
int x, y;
while(~scanf("%d", &p)) {
for(int i=0; i<M; i++) par[i] = i;
memset(under, 0, sizeof(under));
for(int i=0; i<M; i++) sum[i] = 1;
for(int i=0; i<p; i++) {
scanf("%s", cmd);
switch(cmd[0]) {
case 'M':
scanf("%d%d", &x, &y);
merge(x, y);
break;
case 'C':
scanf("%d", &x);
get_par(x);
printf("%d\n", under[x]);
break;
}
}
}
return 0;
}