POJ 1988 Cube Stacking

不正经小新

已于 2023-09-08 08:02:07 修改

阅读量1w

点赞数 1

分类专栏：数据结构与算法文章标签：算法 c++ 并查集后端数据结构

于 2022-03-12 13:41:10 首次发布

本文链接：https://blog.csdn.net/m0_46833224/article/details/123442514

版权

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

63 篇文章 1 订阅

订阅专栏

⭐简单说两句⭐

作者：后端小知识

CSDN个人主页：后端小知识

🔎GZH：后端小知识

🎉欢迎关注🔎点赞👍收藏⭐️留言📝

题目： POJ 1988 Cube Stacking ，哈哈，我们今天来看一道复杂一点的题嘛，这是选自POJ上的一道题，好了，我们一起来看看题意吧：

考虑到直接复制题目，或者截屏的方式不是很方便阅读，我就把直接题目链接放下面！
题目传送门：http://poj.org/problem?id=1988

思路:

读了题目，思考了很久，才发现这道题就是个带权重的并查集,具体就看代码吧！

我们来看看成功AC的代码吧：

#include<iostream>
#include<string>
using namespace std;
int n;
int p[40000];
int cnt[40000];//记录当前节点到根节点的距离（不是树中的距离，而是在栈中，）
int total[40000];//记录树的总节点个数
int find(int x){
    if(x!=p[x]){
        int t=p[x];
        p[x]=find(p[x]);
        cnt[x]+=cnt[t];
    }
    return p[x];
}
void hb(int x,int y){
    int xx=find(x),yy=find(y);
    p[xx]=yy;
    cnt[xx]+=total[yy];
    total[yy]+=total[xx];
}
int main(){
    cin>>n;
    for(int i=0;i<40000;i++){
        p[i]=i;
        cnt[i]=0;
        total[i]=1;
    }
    while(n--){
        string s;   cin>>s;
        if(s=="M"){
            int x,y;    cin>>x>>y;
            hb(x,y);
        }else{
            int x;  cin>>x;
            find(x);
            cout<<cnt[x]<<"\n";
        }
    }
    return 0;
}