HDU 2818 Building Block(带权并查集）

来源：互联网发布：哪个论坛源码好移动版编辑：程序博客网时间：2024/06/02 05:10

题目链接：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 30030typedef 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;}

0 0