POJ1988 CubeStacking (并查集）

本文出自：http://blog.csdn.net/svitter

题意：

开始有N堆方块，编号从1～n。每次移动一堆方块，最后求某个方块下面方块的个数。

输入输出分析：

开始输入一个数字P，代表输入操作个数。

此处发现在g++4.8的版本中，类似与 char ch[0]这样的数组也是可以开辟的。。。

一个不小心开辟了这样一个数组。。然后return 0完全找不到错误所在。。

随后的2～1+p行，每行一组操作数据。

M i j代表移动i的堆到j的堆上。

C i代表求出i以下的方块个数。

数据结构算法分析：

本题目使用并查集来进行计算。

sum
数组来统计当前堆n中方块个数。

under
数组来统计当前n下面的方块个数。

在进行计算的时候，路径压缩和合并均更新under的值。

未进行路径压缩的时候，方块n所记录的under并非final value， 仅仅只是包含了到root
的方块个数。

通过路径压缩的过程，不断的增加当前n的根节点（递归）的under值，求出最终的under值

进行路径合并的时候，更新sum值和under值。当一个堆放在另一个堆上时，被移动的堆的under值一定为0,

此时更新under值为另一个堆的根的sum值，即计算出了此处的部分under值。然后更新合并堆的sum值。

AC代码:

#include <iostream>
#include <stdio.h>
#include <string.h>

using namespace std;

const int MAXN = 30010;

int root[MAXN];
int under[MAXN];
int sum[MAXN];

void init()
{
for(int i = 0; i < MAXN; i++)
{
root[i] = i;
under[i] = 0;
sum[i] = 1;
}
}

int getRoot(int i)
{
if(i == root[i])
return i;
int t = getRoot(root[i]);
under[i] += under[root[i]];
root[i] = t;
return root[i];
}

int a,b;
void Merge(int i, int j)
{
a = getRoot(i);
b = getRoot(j);
if(a == b)
return;
root[a] = b;
under[a] = sum[b];
sum[b] += sum[a];
}

int main()
{
int p;
char ch[0];
int t1, t2;
init();
scanf("%d", &p);
while(p--)
{
scanf("%s", ch);
if(ch[0] == 'M')
{
scanf("%d", &t1);
scanf("%d", &t2);
Merge(t1, t2);
}
else
{
scanf("%d", &t1);
getRoot(t1);
printf("%d\n", under[t1]);
}
}
return 0;
}