POJ 1988 Cube Stacking(带权并查集)

最新推荐文章于 2022-03-03 13:43:07 发布

原创最新推荐文章于 2022-03-03 13:43:07 发布 · 697 阅读

0 ·

CC 4.0 BY-SA版权

数据结构与算法专栏收录该内容

338 篇文章

订阅专栏

本文提供了一种解决POJ1988 CubeStacking问题的方法，通过使用并查集来实现元素合并及查询操作，并详细展示了如何在合并过程中更新元素数量和维护元素间的层级关系。

题目链接：POJ 1988 Cube Stacking

num数组表示某个元素所在组有的元素数目，刚开始在压缩路径时候我还把这一堆的所有元素的num全改变成根节点的num，后来发现并不用这样。above表示某个元素下边有多少的元素，这个数组需要在压缩路径时候维护。

#include <iostream>
#include <cstdio>

using namespace std;

const int MAX_N = 30000 + 100;
int num[MAX_N], p[MAX_N], under[MAX_N];

int _find(int x)
{
    if(p[x] != x)
    {
        int root = _find(p[x]);
        under[x] += under[p[x]];
        //num[x] = num[p[x]];
        return p[x] = root;
    }
    return x;
}
int main()
{
    int P;
    scanf("%d", &P);
    for(int i = 0; i < MAX_N; i++)
        p[i] = i, under[i] = 0, num[i] = 1;
    char str[3];
    int a, b, u, v;
    for(int i = 0; i < P; i++)
    {
       scanf("%s", str);
       if(str[0] == 'M')
       {
           scanf("%d%d", &a, &b);
           u = _find(a);
           v = _find(b);
           if(u != v)
           {
               p[u] = v;
               under[u] += num[v];
               num[v] += num[u];
           }
       }
       else
       {
           scanf("%d", &a);
           _find(a);
           printf("%d\n", under[a]);
       }
    }
    return 0;
}