九度-1017-还是畅通工程

题目描述:
    某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
输入:

    测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。
    当N为0时,输入结束,该用例不被处理。

输出:

    对每个测试用例,在1行里输出最小的公路总长度。

样例输入:
3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0
样例输出:
3
5
来源:

2006年浙江大学计算机及软件工程研究生机试真题

代码(并查集+Kruskal):

#include<stdio.h>
#include<stdlib.h>
#define N 5000
int father[N],rank[N];
void make_set(int x)
{
	father[x]=x;
	rank[x]=0;
}
int find_set(int x)
{
	if(x!=father[x])
		father[x]=find_set(father[x]);
	return father[x];
}
void Union(int x,int y)
{
	x=find_set(x);
	y=find_set(y);
	if(x==y)
		return;
	if(rank[x]>rank[y])
		father[y]=x;
	else
	{
		if(rank[x]==rank[y])
			rank[y]++;
		father[x]=y;
	}
}
struct rood
{
	int x,y;
	int l;
};
int cmp(const void *a,const void *b)
{
	return (*(rood*)a).l>(*(rood*)b).l?1:-1;
}
int main ()
{
	rood a[N];
	int n,i,m,sum,k,x,y;
	while(scanf("%d",&n)!=EOF,n)
	{
		for(i=1;i<=n;i++)
			make_set(i);
		m=n*(n-1)/2;
		for(i=1;i<=m;i++)
			scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].l);
		qsort(a+1,m,sizeof(rood),cmp);
		k=0;sum=0;
		for(i=1;i<=m;i++)
		{
			x=find_set(a[i].x);
			y=find_set(a[i].y);
			if(x!=y)
			{
				Union(a[i].x,a[i].y);
				sum=sum+a[i].l;
				k++;
				if(k==n-1)
					break;
			}
		}
		printf("%d\n",sum);
	}
	return 0;
}


你可能感兴趣的:(最小生成树,并查集)