POJ 1988 Cube Stacking 并查集

（题解来自北大2016 ACM/ICPC竞赛训练暑期课讲义）
题目链接：http://poj.org/problem?id=1988
题意：有N(N<=30,000)堆方块，开始每堆都是一个 方块。方块编号1 – N. 有两种操作：
 M x y ： 表示把方块x所在的堆，拿起来叠放 到y所在的堆上。
 C x : 问方块x下面有多少个方块。
 操作最多有 P (P<=100,000)次。对每次C操 作，输出结果。
解法：
1.除了parent数组，还要开设 sum数组：记录每堆一共有多少方块。
2. 若parent[a] = a, 则sum[a]表示a所在的堆的 方块数目。
3. under数组，under[i]表示第i个方块下面有多少 个方块。 under数组在 堆合并和 路径压缩的时候都要更 新。

#include<iostream>
using namespace std;
#define maxn 31000
int parent[maxn];
int sum[maxn]; //若parent[i]=i sum[i]表示砖块i所在堆的砖块数目
int under[maxn];
int GetParent(int a)
{
    if (parent[a]=a)
        return a;
    int t=GetParent(parent[a]);
    under[a]+=under[parent[a]];
    parent[a]=t;
    return parent[a];
}
void Merge(int a,int b)
{
    int n;
    int pa=GetParent(a);
    int pb=GetParent(b);
    if (pa==pb)
        return;
    parent[pb]=pa;
    under[pb]=sum[pa];  //under[pb] 赋值前一定是0，因为 parent[pb] = pb,pb一定是原b所在堆最底下的
    sum[pa]+=sum[pb];
}
int main()
{
    int p;
    for (int i=0;i<maxn;++i)
    {
        sum[i]=1;
        under[i]=0;
        parent[i]=i;
    }
    cin>>p;
    for (int i=0;i<p;i++)
    {
        char s[20];
        int a,b;
        cin>>s;
        if (s[0]=='M')
        {
            cin>>a>>b;
            Merge(b,a);
        }
        else
        {
            cin>>a;
            GetParent(a);
            cout<<under[a]<<endl;
        }
    }
    return 0;
}