poj 1287 MST(kruskal)

题意:求最小生成树,题目按边输入,有重边

思路:kruskal算法。(题目中说边的数量没有限制,那么应该对重边进行处理。但是没有处理也AC了)

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#define N 10000005
struct edge{
	int x,y,w;
}e[N];
int n,m,father[55];
int cmp(const struct edge* a,const struct edge* b){
	return (*a).w - (*b).w;
}
int find(int x){
	if(father[x] == x)
		return x;
	return father[x] = find(father[x]);
}
int kruskal(){
	int i,x,y,res = 0,edgenum = 0;
	qsort(e,m,sizeof(struct edge),cmp);
	for(i = 1;i<=n;i++)
		father[i] = i;
	for(i = 0;i<m;i++){
		x = find(e[i].x);
		y = find(e[i].y);
		if(x != y){
			res += e[i].w;
			edgenum++;
			father[x] = y;
		}
		if(edgenum == n-1)
			break;
	}
	return res;
}
int main(){
	freopen("a.txt","r",stdin);
	while(scanf("%d",&n)&&n){
		int i;
		scanf("%d",&m);
		for(i = 0;i<m;i++)
			scanf("%d %d %d",&e[i].x,&e[i].y,&e[i].w);
		printf("%d\n",kruskal());
	}
	return 0;
}


你可能感兴趣的:(poj 1287 MST(kruskal))