POJ 1988 Cube Stacking

并查集应用。

题意：初始有N个stack，编号为1-N，给两个操作

M x y  ：将x所在stack的所有cube移到y所在stack的上面

C x     ：统计在x下面的cube的个数

题解：fa[x]表示x的父节点，num[x]表示以x为根的并查集中节点的总数，rank[x]表示x在当前stack（并查集）中的排名

那么，在x下面的cube个数为： num[father(x)] - rank[x] - 1  【father（x）表示x所在并查集的根】

#include <cstdio>
#include <cstring>

const int MAXN = 30003;
int fa[MAXN], num[MAXN], rank[MAXN];

int father(int x) {
    if (x == fa[x]) return x;
    int t = fa[x];
    fa[x] = father(fa[x]);
    rank[x] += rank[t];
    return fa[x];
}
int main() {
    int p, x, y;
    char ch;
    for (int i=1; i<MAXN; i++) {
        fa[i] = i;
        num[i] = 1;
        rank[i] = 0;
    }
    scanf("%d", &p);
    while ( p-- ) {
        scanf(" %c", &ch);
        if (ch == 'M') {
            scanf("%d %d", &x, &y);
            x = father(x);
            y = father(y);
            fa[y] = x;
            rank[y] = num[x];
            num[x] += num[y];
        } else {
            scanf(" %d", &x);
            y = father(x);
            printf("%d\n", num[y]-rank[x]-1);
        }
    }
    return 0;
}