POJ 1988 Cube Stacking 并查集

最新推荐文章于 2020-09-10 10:51:54 发布

原创最新推荐文章于 2020-09-10 10:51:54 发布 · 567 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

ICPC 同时被 3 个专栏收录

52 篇文章

订阅专栏

数据结构

12 篇文章

订阅专栏

Disjoint并查集

9 篇文章

订阅专栏

本文介绍了一道经典的数据结构与算法题目——POJ 1988的解题思路及实现代码。该题涉及并查集、路径压缩等算法技巧，通过维护parent、sum和under三个数组来解决堆叠方块问题。

（题解来自北大2016 ACM/ICPC竞赛训练暑期课讲义）
题目链接：http://poj.org/problem?id=1988
题意：有N(N<=30,000)堆方块，开始每堆都是一个方块。方块编号1 – N. 有两种操作：
 M x y ：表示把方块x所在的堆，拿起来叠放到y所在的堆上。
 C x : 问方块x下面有多少个方块。
 操作最多有 P (P<=100,000)次。对每次C操作，输出结果。
解法：
1.除了parent数组，还要开设 sum数组：记录每堆一共有多少方块。
2. 若parent[a] = a, 则sum[a]表示a所在的堆的方块数目。
3. under数组，under[i]表示第i个方块下面有多少个方块。 under数组在堆合并和路径压缩的时候都要更新。
这里写图片描述

#include<iostream>
using namespace std;
#define maxn 31000
int parent[maxn];
int sum[maxn]; //若parent[i]=i sum[i]表示砖块i所在堆的砖块数目
int under[maxn];
int GetParent(int a)
{
    if (parent[a]=a)
        return a;
    int t=GetParent(parent[a]);
    under[a]+=under[parent[a]];
    parent[a]=t;
    return parent[a];
}
void Merge(int a,int b)
{
    int n;
    int pa=GetParent(a);
    int pb=GetParent(b);
    if (pa==pb)
        return;
    parent[pb]=pa;
    under[pb]=sum[pa];  //under[pb] 赋值前一定是0，因为 parent[pb] = pb,pb一定是原b所在堆最底下的
    sum[pa]+=sum[pb];
}
int main()
{
    int p;
    for (int i=0;i<maxn;++i)
    {
        sum[i]=1;
        under[i]=0;
        parent[i]=i;
    }
    cin>>p;
    for (int i=0;i<p;i++)
    {
        char s[20];
        int a,b;
        cin>>s;
        if (s[0]=='M')
        {
            cin>>a>>b;
            Merge(b,a);
        }
        else
        {
            cin>>a;
            GetParent(a);
            cout<<under[a]<<endl;
        }
    }
    return 0;
}