首页 > 其他 > 详细

最小生成树模板 【kruskal】

时间:2021-04-24 01:18:43      阅读:17      评论:0      收藏:0      [点我收藏+]

算法思想或步骤:(1)将图中的边按照从小到大排序
        (2)按照权值从小到大依次选边。如果当前选取的边使边形成了圈,                    则舍弃;否则标记当前边并计数。(利用并查集判断是否会形成环)
        (3)重复(2)的操作,直到生成树包含n-1条边;否则无法形成最小生成树。

时间复杂度:O(mlogm+m*ɑ(n)) ɑ(n)为一次并查集的复杂度

#include <iostream>
#include <algorithm>
using namespace std;
const int N=5010,M=10010;

int n,m,f[N],ans=0;
struct G{int u,v,w;}e[M];//结构体存图

bool cmp(G a,G b){return a.w<b.w;}

int fa(int u){return f[u]==u? u:f[u]==fa(f[u]);}

void kruskal()
{
	for(int i=1;i<=n;i++) f[i]=i;//初始化并查集
	sort(e+1,e+1+m,cmp);//边排序
	for(int i=1;i<=m;i++){
		int b=fa(e[i].u);
		int c=fa(e[i].v);
		if(b==c) continue;//形成圈,舍弃
		f[c]=b;            
		ans+=e[i].w;       
	}                          
}                                  
                                  
int main()
{
	ios::sync_with_stdio(false);
	cin>>n>>m;
	int a,b,c;
	for(int i=1;i<=m;i++){
		cin>>a>>b>>c;
		e[i].u=a,e[i].v=b,e[i].w=c;
	}
	kruskal();
	
	cout<<ans<<endl;
	return 0;
}

最小生成树模板 【kruskal】

原文:https://www.cnblogs.com/unravel-CAT/p/14695073.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!