POJ 1988 Cube Stacking(带权并查集)

题目链接：POJ 1988 Cube Stacking

num数组表示某个元素所在组有的元素数目，刚开始在压缩路径时候我还把这一堆的所有元素的num全改变成根节点的num，后来发现并不用这样。above表示某个元素下边有多少的元素，这个数组需要在压缩路径时候维护。

#include <iostream>
#include <cstdio>

using namespace std;

const int MAX_N = 30000 + 100;
int num[MAX_N], p[MAX_N], under[MAX_N];

int _find(int x)
{
    if(p[x] != x)
    {
        int root = _find(p[x]);
        under[x] += under[p[x]];
        //num[x] = num[p[x]];
        return p[x] = root;
    }
    return x;
}
int main()
{
    int P;
    scanf("%d", &P);
    for(int i = 0; i < MAX_N; i++)
        p[i] = i, under[i] = 0, num[i] = 1;
    char str[3];
    int a, b, u, v;
    for(int i = 0; i < P; i++)
    {
       scanf("%s", str);
       if(str[0] == 'M')
       {
           scanf("%d%d", &a, &b);
           u = _find(a);
           v = _find(b);
           if(u != v)
           {
               p[u] = v;
               under[u] += num[v];
               num[v] += num[u];
           }
       }
       else
       {
           scanf("%d", &a);
           _find(a);
           printf("%d\n", under[a]);
       }
    }
    return 0;
}