Kruskal算法

#110. 最小生成树
http://219.153.61.2:9000/problem/110
题目描述

输入一个连通无向网络，输出它的最小生成树的权值和

输入

输入文件第一行为n和m（n,m<=20），表示n个顶点和m条边，接下来m行，每行vi，vj，k(0 < k <=100)，表示vi到vj有条边，而且权值是k

输出

输出文件仅一行，为最小生成树的权值和

输入样例
4 5
1 2 2
1 3 1
2 3 5
3 4 2
4 1 3

输出样例
5

#include<bits/stdc++.h>
using namespace std;
#define N 10010
struct node{int x,y,w;}e[N];
int n,m,up[N],mst,idx;
int find(int x){return (up[x]==x)?up[x]:find(up[x]);}
bool join(int x,int y){
	int a=find(x),b=find(y);
	if(a!=b){
		up[a]=b;
		return 1;
	}
	return 0;
}
bool comp(node a,node b){return a.w<b.w;}
int main(){
	scanf("%d%d",&n,&m);
	int x,y,z;
	for(int i=1;i<=m;i++){
		scanf("%d%d%d",&x,&y,&z);
		e[++idx].w=z;e[idx].y=y;e[idx].x=x;//建立边集数组
	}
	for(int i=1;i<=n;i++)up[i]=i;//初始化并查集数组
	sort(e+1,e+m+1,comp);//边集数组排序
	for(int i=1;i<=idx;i++)
		if(join(e[i].x,e[i].y))//判断是否形成回路
			mst+=e[i].w;//累加最小生成树权值
	printf("%d",mst);
	return 0;
}