POJ 1988 Cube Stacking 并查集

#include <iostream>
#include <cstdio>
using namespace std;

int n;
int f[30020];
int sum[30020];
int num[30020];

int getf(int a)
{
    if(f[a] == a)
    {
        return a;
    }
    int t = f[a];
    
    f[a] = getf(f[a]);
    
    num[a] += num[t];
    
    return f[a];
}

void merge(int a, int b)
{
    int x = getf(a);
    int y = getf(b);
    
    if(x != y)
    {
        f[y] = x;
        num[y] += sum[x];
        sum[x] += sum[y];
    }
}

void init()
{
    for(int i=1; i<=30010; i++)
    {
        f[i] = i;
        num[i] = 0;
        sum[i] = 1;
    }
}

int main()
{
    char s[2];
    int a,b; 
    while(cin>>n)
    {
        init(); 
        for(int i=1; i<=n; i++)
        {
            scanf("%s",s);
            if(s[0] == ‘M‘)
            {
                scanf("%d %d",&a, &b);
                merge(a,b);
            }
            else if(s[0] == ‘C‘)
            {
                scanf("%d",&a);
                printf("%d\n", sum[getf(a)]-num[a]-1);
            }
        }
    }
    return 0;
}