HDU 2818 Building Block(带权并查集）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2818

题       意：有1----n个堆，每一堆有一个blocks，并且对他们进行两个操作：

                             1、把 X 所在的那一堆放到 Y 所在的那一堆上面。

                         2、询问 X 下面有多少个blocks。

思       路：使用并查集处理数据，并且将每个结点的高度与它之下的blocks数记录下来。

                    注：需要使用状态压缩，否则会超时。

 

代码如下：

#include <iostream>
using namespace std;
#include <string.h>
#include <stdio.h>
#include <queue>
#include <algorithm>
#define m 30030
typedef long long LL;
int ben[m],high[m],under[m];
int findx(int x)//状态压缩后的find函数
{
    if(x==ben[x]) return x;
    int z = findx(ben[x]);
    under[x] += under[ben[x]];
    return ben[x]=z;
}
void mergex( int x, int y )
{
    int fx = findx(x);
    int fy = findx(y);
    if(fx!=fy)
    {
        under[fx]=high[fy];//记录该结点原有的blocks数
        ben[fx]=fy;
        high[fy]+=high[fx];//初始化blocks数目
    }
}
int main()
{
    int T;
    scanf ( "%d", &T );
    for( int i = 0; i <= m; i ++ )
        ben[i]=i,under[i]=0,high[i]=1;
    while ( T-- )
    {
        char ch;
        scanf ( " %c", &ch );
        if( ch == 'M' )
        {
            int x,y;
            scanf ( "%d %d", &x, &y );
            mergex(x,y);
        }
        else if( ch == 'C' )
        {
            int x;
            scanf ( "%d", &x );
            findx(x);
            printf("%d\n",under[x]);
        }
    }
    return 0;
}