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;
}