POJ 1988 Cube stacking

题目如下：
Sample Input 6 M 1 6 C 1 M 2 4 M 2 6 C 3 C 4 
Sample Output 1 0 2
有N(N<=30,000)堆方块，开始每堆都是一个方块
方块编号1 – N. 有两种操作
M x y ：表示把方块x所在的堆，拿起来叠放到y 所在的堆上
C x : 问方块x下面有多少个方块
操作最多有 P (P<=100,000)次
对每次C操作，输 出结果

基本思路是维护两个数组

under数组表示某个节点下面的数目
sum表示某个节点所在的方块总和，只用维护根节点即可

#include<iostream>
using namespace std;
int sum[3000] = {};//结点加上自己上的和，只维护根节点
int under[3000] = {};//节点下的数目，路径压缩的问题
int parent[3000] = {};
int n;
int getroot(int a)
{
	if (parent[a] == a)
	{
		return a;
	}
	int t = getroot(parent[a]);
	under[a] += under[parent[a]];//路径压缩时要把parent的under值层层加到结点的under上
	parent[a] = t;
	return parent[a];
}
void merge(int a,int b)//b摞到a上
{
	int aa = getroot(a);
	int bb = getroot(b);//寻根加路径压缩
	if (aa == bb)
	{
		return;
	}
	under[bb] = sum[aa];//bb为根，underbb一定为0，直接加
	sum[aa] += sum[bb];
	parent[bb] = aa;
}
void init()
{
	for (int i = 0; i <= 100; i++)
	{
		parent[i] = i;
		sum[i] = 1;//开始每个节点都是只有自己一个
	}
}
int main()
{


	cin >> n;
	init();

	for (int i = 0; i < n; i++)
	{
		char op;
		cin >> op;

		if (op == 'M')
		{
			int a, b;
			cin >> a >> b;
			merge(b, a);
		}
		else if (op == 'C')
		{
			int C;
			cin >> C;
			getroot(C);
			cout << under[C] << endl;;
		}
		else
		{
			return 0;
		}
	}

	return 0;
}