最小生成树——Kruskal算法

#include<stdio.h>
#include<algorithm>
using namespace std;

const int maxe=1010;
const int maxv=110;

struct edge{
	int u,v;	//边的两个端点编号 
	int cost;	//边权 
}e[maxe];

bool cmp(edge a,edge b){
	return a.cost<b.cost;
}

int father[maxv];
int findFather(int x){
	int a=x;
	while(x!=father[x])
		x=father[x];
	//路径压缩
	while(a!=father[a]){
		int z=a;
		a=father[a];
		father[z]=x;
	} 
	return x;
}

int kruskal(int n,int m){	//顶点数为n边数为m 
	int ans=0,num_edge=0;
	int fatheru,fatherv;
	sort(e,e+m,cmp);
	for(int i=0;i<n;i++)
		father[i]=i;
	for(int i=0;i<m;i++){
		fatheru=findFather(e[i].u);
		fatherv=findFather(e[i].v);
		if(fatheru!=fatherv){	//如果不在一个集合中 
			father[fatheru]=fatherv;	//合并集合 
			ans+=e[i].cost;		//边权之和增加测试边的边权 
			num_edge++;			//当前生成树的边数加1 
		}
		if(num_edge==n-1)
			break;
	}
	if(num_edge!=n-1) return -1;
	else return ans;	//返回最小生成树的边权之和 
}

int main(){
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=0;i<m;i++){
		scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].cost);
	}
	int ans=kruskal(n,m);	//kruskal算法入口 
	printf("%d\n",ans);
	return 0;
}