本文出自: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]数组来统计当前堆n中方块个数。
under[n]数组来统计当前n下面的方块个数。
在进行计算的时候,路径压缩和合并均更新under的值。
未进行路径压缩的时候,方块n所记录的under并非final value, 仅仅只是包含了到root[n]的方块个数。
通过路径压缩的过程,不断的增加当前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;
}