并查集 POJ 1988 Cube Stacking

题目传送门:     -------->这里<----------

题目大意：

　　有标记1-N的方块，初始时一个砖块是一堆。然后进行以下操作

　　M X Y ： 将 X 砖块所在堆 叠到 Y 砖块所在堆上面；

　　C X ：数在 X 砖块所在堆中 叠在X砖块下的砖块个数；

　　1<=N<=30000;

思路：

　　并查集，每一个砖块堆是一个集合。将每一堆的最上面一块砖设为根，合并时更新 Y 堆的根的 father 和 val (在Y上面的方块数) 。 要达到这些目的，需要一个all数组，来记录每一堆里的砖块总数，和一个val数组，来记录具体某一块砖上面的砖块个数。输出时，输出all － val － 1；在find函数中进行路径压缩的，记住回溯时将val值相加。

初始时   all 数组 全为 1 ，不能用memset初始化int 数组全为1，因为memset是按字节填充，会使值错误。

　　　　val 数组 全为 0。

　　　　father[x] = [x];

代码：

#include <iostream>
#include <stdio.h>
#include <string.h>
#define MAX 30010

using namespace std;
int val[MAX];
int all[MAX];
int father[MAX];
void init()
{
    memset(val,0,MAX * sizeof(int));
    for(int i = 0 ; i < MAX ; i++) 
    {
        father[i] = i;
        all[i] = 1;
    }
    return ;
}
int Find_Root(int x)
{
    if(father[x] == x)
    {
        return x;
    }
    int temp = father[x];
    father[x] = Find_Root(father[x]);
    val[x] += val[temp];
    return father[x];
}


void M (int a ,int b)
{
    if(a == b) return ;
    father[a] = b;
    val[a] += all[b];
    all[b] += all[a];
    return ;
}

void C(int a)
{
    cout << all[Find_Root(a)] - val[a] - 1 << endl;
    return ;
}


int main()
{
    int T;
    while(~scanf("%d",&T))
    {
        init();
        for (int i = 0; i < T; ++i)
        {
            char op;
            int n,m;
            cin >> op ;
            if(op == 'M')
            {
                cin >> n >> m;
                M(Find_Root(m),Find_Root(n));
            }
            else if(op == 'C')
            {
                cin >> n;
                C(n);
            }
            //////////////////////////////
        }
    }
    return 0;

}