图论之cruskal算法(克鲁斯卡尔)

我们已经学了prim算法了,接下来我们来学一下cruskal算法,和prim算法不同的点就在于prim是不断的加结点,而cruskal是不断的加边,不断的加最小的边,我们需要把每个边的权值用结构体存起来,然后排序,从小到大遍历边,不断的加边

如图,我们从权值最小的边开始

当我们把1,2权值为2的边加上去的话,就不符合我们要找生成树的性质了,

那我们怎么维护生成树呢?我们可以把形成生成树的这些结点都放在一个集合里,然后接下来看插入的边的两个结点是不是位于生成树集合里,如果位于生成树集合的话就不连了

OK,那么废话不多说,我们来实现一下代码吧

cpp 复制代码
#include <iostream>
#include <algorithm>
using namespace std;
int n, m;
const int N = 2e5 + 10, INF = 0x3f3f3f3f;
struct node {
	int x;//结点1
	int y;//结点2
	int z;//权值 
}a[N];
int fa[N];
int find(int x)
{
	if (fa[x] == x) return x;
	return fa[x] = find(fa[x]);
}
void un(int x, int y)
{
	int px = find(x); int py = find(y);
	fa[px] = py;
}
bool cmp(const node& x, const node& y)
{
	return x.z < y.z;
}
int ret = 0;
int cnt = 0;//记录加入了几条边 
int kk()
{
	sort(a + 1, a + 1 + m, cmp);
	for (int i = 1; i <= m; i++)
	{
		int x1 = a[i].x, y1 = a[i].y;
		int z1 = a[i].z;
		int fx = find(x1), fy = find(y1);
		if (fx != fy)
		{
			ret += z1;
			cnt++;
			un(fx, fy);
		}

	}
	return cnt == n - 1 ? ret : INF;
}
int main()
{
	cin >> n >> m;
	for (int i = 1; i <= n; i++)
	{
		fa[i] = i;
	}
	for (int i = 1; i <= m; i++)
	{
		cin >> a[i].x >> a[i].y >> a[i].z;
	}
	int r = kk();
	if (r == INF) cout << "orz" << endl;
	else cout << r << endl;




	return 0;
}
相关推荐
国科安芯19 分钟前
星上计算的第一道防线:低轨遥感卫星载荷处理链中的AS32S601型抗辐射MCU—— 面向高分辨率成像与在轨智能处理的场景化综述
大数据·网络·人工智能·单片机·算法·低轨卫星·抗辐射
thusloop29 分钟前
零门槛接入 GPT-6-astra 与 claude-opus-5-5,三大场景实战指南
python·gpt·算法·agent·claude·codex
瓦瓦gogo44 分钟前
学回溯算法时,我先弄明白了怎么“退回来”
算法
Lzg_na1 小时前
堆和堆排序
算法
TK泰妞1 小时前
TikTok 带货视频如何做多语言本地化?从翻译脚本到重做场景的实用方法
算法·音视频
房开民1 小时前
DDPM 完整原理 + 全套公式串联
算法
lunar_coder1 小时前
哈希扩展 -- 位图思想
c++·算法·哈希算法
秋天的怀念_David1 小时前
听故事学语言 · 数据结构篇 · 序章一 算法之问
数据结构·程序人生·算法
LG.YDX2 小时前
基于中心的3D目标检测和探索追踪-笔记
笔记·算法·目标检测
民乐团扒谱机2 小时前
【读论文】音色在微分音程记忆中的作用
人工智能·深度学习·算法