-
题目描述:
-
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
-
输入:
-
测试输入包含若干测试用例。每个测试用例的第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
-
来源:
代码(并查集+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;
}