某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
当N为0时,输入结束,该用例不被处理。
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
Huge input, scanf is recommended.
程序如下:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int pre[100001];
struct node
{
int u,v;
int value;
}p[100001];
void init(int n)
{
for(int i=0;i<n;i++)
{
pre[i]=i;
}
}
int find(int x)
{
if(pre[x]==x) return x;
return pre[x]=find(pre[x]);
}
bool cmp (node x,node y)
{
return x.value<y.value;
}
void mix(int x,int y)
{
int fx=find(x);
int fy=find(y);
if(fx!=fy)
{
pre[fy]=fx;
}
}
int main()
{
int n;
while(scanf("%d",&n)!=EOF&&n!=0)
{
memset(pre,0,sizeof(pre));
int m=n*(n-1)/2;
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&p[i].u,&p[i].v,&p[i].value);
}
sort(p,p+m,cmp);
int num=0,ans=0;
init(n);
for(int i=0;i<m&&num<n-1;i++)
{
int fx=find(p[i].u);
int fy=find(p[i].v);
if(fx!=fy)
{
pre[fy]=fx;
num++;
ans+=p[i].value;
}
}
printf("%d\n",ans);
}
return 0;
}
原文章网址:点击打开链接