C++并查集模板代码(路径压缩和代码优化)

题目描述
在某个城市中住着n个人,现在给定关于这n个人的m条信息(即某2个人认识)。
假设所有认识的人一定属于同一个单位,请计算该城市有多少个单位?
输入
第1行的第1个值表示总人数n,第2个值表示总信息数m;第2行开始为具体的认识关系信息
输出
单位的个数
样例输入
10 4
2 3
4 5
4 8
5 8
样例输出
7

个人分析:
之前写过关于并查集的题目,题目名我还记得how many tables 上次还只是初始并查集,参考了别人举的倚天屠龙记的例子了解了一下,但这次学习了并查集的路径压缩 时间上更加优化了

erank数组就是记录以每个顶点为根的秩,然后将矮的书的根连上高的树,这样高的树深度不会发生改变 比如说 2 给到3 高度还是3 不会发生改变 也是为了我们后面找根进行优化,只需要O(1)的时间就可以找到 相当于很多个顶点连上一个根,这样我们找根就方便多了
核心代码:

//路径压缩
int find_pre(int x)
{
    if(x!=pre[x])
    {
        pre[x]=find_pre(pre[x]);
    }
    else
    {
        return pre[x];
    }
}

AC代码:

#include<bits/stdc++.h>
using namespace std;
const int maxn=1000+8;
int pre[maxn];
int erank[maxn];
int n,m;
//路径压缩
int find_pre(int x)
{
    if(x!=pre[x])
    {
        pre[x]=find_pre(pre[x]);
    }
    else
    {
        return pre[x];
    }
}

void join(int x,int y)
{
    int a=find_pre(x);
    int b=find_pre(y);
    if(erank[a]<erank[b])
    {
        pre[a]=b;
    }
    else
    {
        pre[b]=a;
        erank[a]++;
    }
}

int main()
{

    while(cin>>n>>m)
    {
        for(int i=1;i<=n;i++)
        {
            pre[i]=i;
            erank[i]=0;
        }
        for(int i=0;i<m;i++)
        {
            int x,y;
            cin>>x>>y;
            join(x,y);
        }
        int res=0;
        for(int i=1;i<=n;i++)
        {
            if(pre[i]==i)
            {
                ++res;
            }
        }
        cout<<res<<endl;
    }

    return 0;
}

学如逆水行舟,不进则退

在这里插入图片描述

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

一百个Chocolate

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值