POJ 1988 Cube Stacking

题目分析

典型的带权并查集，这道题有2个操作，一个是将带a的那一堆石头移到带b的那一堆石头上面，另一个是询问a号石头下面有多少石头。这里我定义了3个数组，fa用于查父节点，he查编号为i的石头下面有多少石头，sum表示以a为根节点的石头堆中有多少石头。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 30005;

int sum[maxn],he[maxn],fa[maxn];  //sum表示这堆石头有多少个，he表示

void init(){
    for(int i = 0; i < maxn; i++){
        sum[i] = 1;
        he[i] = 0;
        fa[i] = i;
    }
}

int Find(int x){
    if(x == fa[x]) return fa[x];
    int temp = Find(fa[x]);
    he[x] += he[fa[x]];
    fa[x] = temp;
    return fa[x];
}

int main()
{
    int P;
    scanf("%d", &P);
    char op;
    int x,y;
    init();
    while(P--){
        cin >> op;
        if(op == 'M') {
            scanf("%d%d", &x, &y);
            int xx = Find(x);
            int yy = Find(y);
            he[xx] = sum[yy];
            sum[yy] += sum[xx];
            fa[xx] = yy;
        }else{
            scanf("%d", &x);
            Find(x);
            printf("%d\n", he[x]);
        }
    }
    return 0;
}