HDU 1233 还是畅通工程 (并查集)

#include <bits/stdc++.h>
using namespace std;
const int maxn=10005; 
struct node
{
	int f;
	int l;
	int len;
}a[maxn];//定义一个结构体储存公路的信息 
int n,m,b[maxn],ans;
bool cmp(node a,node b)
{
	return a.len<b.len;//按公路的长度从小到大排序 
}
int find(int x)
{
	int r=x;
	while(b[r]!=r)//寻找父节点 
	r=b[r];//如果当前部位父节点则继续寻找 
	return r;//返回父节点 
}
void mergy(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
	b[fy]=x;//将父节点合并 
}
int kruskal()
{
	sort(a,a+m,cmp);//排序，优先查找长度小的 
	for(int i=0;i<=n;i++)
	b[i]=i;//给b数组赋值 
	ans=0;//长度总和 
	for(int i=0;i<m;i++)
		if(find(a[i].f)!=find(a[i].l))//如果两个村子没有公路 
		{
			mergy(a[i].f,a[i].l);//查找父节点 
			ans+=a[i].len;//长度相加	
		}
		return ans;
}
int main()
{
	while(scanf("%d",&n)!=EOF&&n!=0)
	{
		m=n*(n-1)/2;
		for(int i=0;i<m;i++)
		scanf("%d %d %d",&a[i].f,&a[i].l,&a[i].len);		
		printf("%d\n",kruskal());
	}
	return 0;
}