POJ1988 Cube Stacking

　　原题传送：http://poj.org/problem?id=1988

　　一开始的想法是把下堆连到上堆去（这很符合树的结构思想），然后另外开一个数组标记录节点的值，递归的时候字节点加上这个标记的值与根节点的差值，但是发现，这样做在某些union_set时会使同一个根的所有节点标记的值错乱。

　　这道题需要对并查集有比较深刻的理解，多开一个数组记录堆总数。把上堆union到下堆上去，这样，总的根就是下堆的根，最上堆总是在最下层，find_set的时候就很方便求出答案。

#include <stdio.h>
#define N 30001

int a[N], f[N], num[N], v[N];

int find_set(int x)
{
    if(x != f[x])
    {
        int tmp = f[x];
        f[x] = find_set(f[x]);
        num[x] += num[tmp];
    }
    return f[x];
}

void union_set(int a, int b)
{
    int x = find_set(a);
    int y = find_set(b);
    if(x != y)
    {
        f[x] = y;
        num[x] += v[y];
        v[y] += v[x];
    }
}

void make_set()
{
    for(int i = 0; i < N; i ++)
        f[i] = i, num[i] = 0, v[i] = 1;
}

int main()
{
    int p, a, b;
    char op[5];
    make_set();
    scanf("%d", &p);
    while(p --)
    {
        scanf("%s", op);
        if(op[0] == 'M')
        {
            scanf("%d%d", &a, &b);
            union_set(a, b);
        }
        else
        {
            scanf("%d", &a);
            find_set(a);
            printf("%d\n", num[a]);
        }
    }
    return 0;
}