最小生成树 kruskal算法

算法思路

  1. 将图中的所有边都去掉
  2. 将边按权值从小到大的顺序添加到图中,保证添加的过程中不会形成环
  3. 重复上一步直到连接所有顶点,此时就生成了最小生成树

采用了贪心策略

代码

代码中用了并查集,可以查看并查集

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

struct edge									//边 
{
	int from,to,dis;
}e[200001];
int fa[100001];				 				//父节点 
int n,m,cnt,ans;

bool cmp(edge a,edge b)
{
	return a.dis<b.dis;						//用于排序 
}

int getfa(int i)							//寻找根节点 
{
	if(fa[i]==i)
		return i;
	return getfa(fa[i]);
} 

void unionn(int u,int v)
{
	fa[getfa(u)]=getfa(v);					//合并 
}

void kruskal()
{
	for(int i=1;i<=n;i++)
		fa[i]=i;							//先将自己设为自己的父节点
	sort(e+1,e+1+m,cmp);					//排序 
	for(int i=1;i<=m;i++)
	{
		if(cnt==n-1)						//共n-1条边 
			return;
		if(getfa(e[i].from)!=getfa(e[i].to))//如果根节点相同,则说明已经成环 
		{
			unionn(e[i].to,e[i].from);		//使用一条边后,将两端点设置为子节点和父节点 
			ans+=e[i].dis;
			cnt++;
		}
	}
}

int main()
{
	cin>>n>>m;
	for(int i=1;i<=m;i++)
		scanf("%d%d%d",&e[i].from,&e[i].to,&e[i].dis);
	kruskal();
	cout<<ans;
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值