最小生成树(MST)的Kruskal算法

内容来源 紫皮书p356 kruskal算法的思想在于合并


Kruskal算法的第一步是给所有边按照从小到大的顺序排列。这一步可以直接使用sort。接下来从小到大依次考察每条边(u,v)。
u,v如果在同一个连通分量中,那么加入(u,v)后会成环

伪代码:

把所有边排序,记第i小的边为e[i];
初始化MST为空
初始化连通分量,让每个点自成一个独立的连通分量
for(int i = 0; i < m; i++)
	if(e[i].u和e[i].v不在同一个连通分量) {
   
			把边e[i]加入MST
			合并e[i].u和e[i].v所在的连通分量
	}

最关键的在于连通分量的查询和合并
需要知道任意两个点是否在同一个连通分量中
还需要合并两个分量
因此要使用并查集,把每个连通分量看成一个集合,每个点恰好属于一个连通分量,图的所有连通分量可以用若干不相交集合表示

Kruskal的完整代码
假设第i条边的两个端点序号和权值分别储存在u[i],v[i],w[i]中

int cmp(const int i, const int j) {
   return w[i]<w[j]; }//间接排序
int find(int x) {
   return p[x] == x ? x : p[x] = find(p[x]);}
int Kruskal() {
   
	int ans = 0;
	for(int i = 0; i < n; i++) p[i] = i;//初始化并查集
	for(int i = 0; i < m; i++) r[i] = i;//初始化边序号
	sort(r,r+m,cmp);
	for(int i = 0; i < m; i++) {
   
		int e = r[i]; int x = find(u[e]); int y = find(v[e]);
		if(x != y) {
   ans += w[e]; p[x] = y;}
	}
	return ans;
}
全部评论

相关推荐

totoroyyw:千年老妖😂
投递华为等公司10个岗位
点赞 评论 收藏
分享
孤寡孤寡的牛牛很热情:为什么我2本9硕投了很多,都是简历或者挂,难道那个恶心人的测评真的得认真做吗
点赞 评论 收藏
分享
评论
1
收藏
分享
牛客网
牛客企业服务