HDU-2818-Building Block（基础带全并查集每次移动一幢木块求该木块下方的个数）

最新推荐文章于 2019-05-28 15:46:00 发布

Helium_wild

最新推荐文章于 2019-05-28 15:46:00 发布

阅读量244

点赞数

CC 4.0 BY-SA版权

分类专栏：图论==并查集

本文链接：https://blog.youkuaiyun.com/qq_42576687/article/details/88298608

图论==并查集专栏收录该内容

7 篇文章

订阅专栏

本文详细解析了一款积木游戏背后的算法实现。通过使用并查集数据结构，文章阐述了如何处理积木堆叠和计算特定积木下块数的问题。关键在于维护一个并查集，用于快速查找和合并积木堆，以及两个辅助数组cnt[]和sum[]，分别存储答案和堆内积木数量。此算法能高效地处理大量操作，适用于竞赛编程和游戏开发。

题目
约翰正在玩积木。有N个block (1 <= N <= 30000)编号为1…N .初始有N个桩，每桩包含一个block。然后John做一些操作P次(1 <= P <= 1000000)。有两种操作:
M X Y:将包含X块的整桩放到包含Y块的桩上。如果X和Y在同一桩中，则忽略此命令。
C X:计算X块下的块数
要求您找出每个C操作的输出。

输入
第一行包含整数P，然后是P行，每一行都包含上面描述的操作。

输出
在一行中输出每个C操作的计数。

题解：不仅需要cnt[ ]数组存储答案,还需要sum[ ]数组维护。合并的时候需要用到。
ans[ra]=sum[rb],sum[rb]+=sum[ra]。

#include <cstdio>
using namespace std;
const int N=300000+5;
int f[N],ans[N],sum[N];
int seek(int x){
    if(x==f[x]) return x;
    int fa=f[x];
    f[x]=seek(f[x]),ans[x]+=ans[fa];
    return f[x];
}
int main(){
    int n;
    while(~scanf("%d",&n)){
        getchar();
        for(int i=1;i<=n;i++) f[i]=i,ans[i]=0,sum[i]=1;
        while(n--){
            char ch;scanf("%c",&ch);
            if(ch=='M'){
                int a,b;scanf("%d%d",&a,&b);++a,++b;
                int ra=seek(a),rb=seek(b);
                if(ra!=rb) f[ra]=rb,ans[ra]=sum[rb],sum[rb]+=sum[ra];
            }
            else{
                int x;scanf("%d",&x);++x,seek(x);
                printf("%d\n",ans[x]);
            }
        }
    }
}