escription
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。
当N为0时,输入结束,该用例不被处理。
Output
对每个测试用例,在1行里输出最小的公路总长度。
Sample Input
8
1 2 42
1 3 68
1 4 35
1 5 1
1 6 70
1 7 25
1 8 79
2 3 59
2 4 63
2 5 65
2 6 6
2 7 46
2 8 82
3 4 28
3 5 62
3 6 92
3 7 96
3 8 43
4 5 28
4 6 37
4 7 92
4 8 5
5 6 3
5 7 54
5 8 93
6 7 83
6 8 22
7 8 17
0
Sample Output
82
思路
最小生成树模板题
代码
#include<bits/stdc++.h>
#define MAX 10005
using namespace std;
struct edge{
int u;
int v;
int w;
}e[MAX];
bool cmp(edge x,edge y)
{
return x.w<y.w;
}
int n,m,f[MAX],sum,c;
int getf(int v)
{
if(f[v]==v)
return v;
else
{
f[v]=getf(f[v]);
return f[v];
}
}
int Merge(int v,int u)
{
int t1=getf(v);
int t2=getf(u);
if(t1!=t2)
{
f[t2]=t1;
return 1;
}
return 0;
}
void init()
{
for(int i=1;i<=n;i++)
{
f[i]=i;
}
}
int main()
{
int i;
while(~scanf("%d",&n))
{
if(n==0) break;
if(n==1)
{
printf("0\n");
continue;
}
int m=n*(n-1)/2,x;
for(i=1;i<=m;i++)
{
scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
}
sort(e+1,e+m+1,cmp);
init();
c=sum=0;
for(i=1;i<=m;i++)
{
if(Merge(e[i].u,e[i].v))
{
c+=1;
sum+=e[i].w;
}
if(c==n-1)
break;
}
printf("%d\n",sum);
}
return 0;
}