最小生成树(kruskal算法)

#include <cstdio>
#include <iostream>
#include <queue>
#include <vector>
#include <cstring>
#include <algorithm>
#include <string>
#include <map>
#include <set>
using namespace std;
int n,m;
const int maxn = 1e5;
int f[maxn];
struct node
{
    int x;int y;int d;
}tmp[maxn];
int cmp(node a,node b)
{
    return a.d < b.d;
}
int getf(int x)
{
    if(f[x] == x) return x;
    else  return (f[x] = getf(f[x]));
}
int mix(int x,int y)
{
    int t1 = getf(x);
    int t2 = getf(y);
    if(t1 != t2)
    {
        f[t1] = t2;
        return 1;
    }
    return 0;
}
int main()
{
    cin >> n >> m;
    for(int i = 0;i <= n;i++)
        f[i] = i;
    for(int i = 0;i < m;i++)
        cin >> tmp[i].x >> tmp[i].y >> tmp[i].d;
    sort(tmp,tmp+m,cmp);
    long long sum = 0;
    for(int i = 0;i < m;i++)///算法的核心部分
    {
        if(mix(tmp[i].x,tmp[i].y))
            sum += tmp[i].d;
    }
    cout << sum << endl;
    return 0;
}
/**
6 9
2 4 11
3 5 13
4 6 3
5 6 4
2 3 6
4 5 7
1 2 1
3 4 9
1 3 2
**/

算法的核心部分其实还可以优化,就是当边检验到n-1的时候,就已经成了一棵树了,就不需要继续进行判断的,可以省下一部分判断的时间

long long sum = 0;
 int cnt  = 0;
    for(int i = 0;i < m;i++)
    {
        if(mix(tmp[i].x,tmp[i].y))
            sum += tmp[i].d,cnt++;
            if(cnt == n-1)
                break;
    }



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值