最小生成树模板——Kruskal算法

版权声明：本文为博主原创文章，欢迎转载。如有问题，欢迎指正。 https://blog.csdn.net/weixin_42172261/article/details/87165842

Kruskal算法优先试用于稀疏图
把n个点的中m条边的边权按照从小到大排序，然后遍历这m条边，如果这一条边两端的两个不连通，就合并，如果连通了，这条边就作废（这里用到了并查集的知识）。一直到用到的边数为n-1条，因为生成一个树只需要n-1条边。

代码：

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

struct edge{
	int u;
	int v;
	int w;
}e[105];
int n, m, pre[105]; 

int find(int x) //并查集模板 
{
	int r=x;
	while (pre[r]!=r){
		r=pre[r];
	} 
	
	int i=x, j;
	while (i!=r){
		j=pre[i];
		pre[i]=r;
		i=j; 
	}
	
	return r;
} 
void join(int x, int y) 
{
	int fx=find(x), fy=find(y);
	if (fx!=fy)
		pre[fx]=fy;
} 

int cmp(edge a, edge b)
{
	return a.w<b.w;
}

int main()
{
	int	sum=0, count=0;
	scanf("%d%d", &n, &m);
	for (int i=1; i<=m; i++)
		scanf("%d%d%d", &e[i].u, &e[i].v, &e[i].w);
		
	sort(e+1, e+m+1, cmp);//边权由小到大排序 
	for (int i=1; i<=n; i++)//每个点一开始相互独立 
		pre[i]=i;
		
	for (int i=1; i<=m; i++){
		if (find(e[i].u)!=find(e[i].v)){
			join(e[i].u, e[i].v);
			count++;
			sum+=e[i].w;
		}
		if (count==n-1)//n-1条边构成树 
			break;
	}
	printf("%d", sum);//sum为树的最小边权之和 
	return 0;
}