还是畅通工程
Problem Description
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。 当N为0时,输入结束,该用例不被处理。
Output
对每个测试用例,在1行里输出最小的公路总长度。
Sample Input3
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
Sample Output
3
5
一.普利姆算法(Prim)
时间复杂度为O(n^2),其中n为顶点的个数;它的执行时间与图中的边数无关,适合于稠密图。
#include<iostream>
#include<string.h>
using namespace std;
#define inf 0x3f3f3f3f
int vis[1007],dis[1007],n,m;
int pre[1007][1007];
void prime()
{
int i,j,k;
int idex=0;
int ans=0;
memset(vis,0,sizeof(vis));
vis[1]=1;
for(i=2; i<=n; i++)
dis[i]=pre[1][i];
for(i=2; i<=n; i++)
{
int min=inf;
idex=0;
for(j=1; j<=n; j++)
{
if(!vis[j]&&dis[j]<min)
{
min=dis[j];
idex=j;
}
}
ans+=dis[idex];
vis[idex]=1;
for(k=1; k<=n; k++)
{
if(!vis[k]&&dis[k]>pre[idex][k])
dis[k]=pre[idex][k];
}
}
cout<<ans<<endl;
}
int main()
{
while(cin>>n&&n)
{
m=n*(n-1)/2;
int i,u,v,w;
memset(vis,0,sizeof(vis));
memset(pre,inf,sizeof(pre));
for(i=1; i<=m; i++)
{
cin>>u>>v>>w;
pre[u][v]=w;
pre[v][u]=w;
}
prime();
}
return 0;
}
二.克鲁斯卡尔算法(Kruskal)
时间复杂度为(nlog2n),其中n为边的个数;他的执行时间与图中边数有关,适合于稀疏图。
#include <iostream>
#include <string.h>
#include <math.h>
#include <algorithm>
using namespace std;
struct node
{
int u,v,w;
};
node arr[10000];
int pre[110];
int n;
bool cmp(node a,node b)
{
return a.w<b.w;
}
int find(int x)
{
if(x==pre[x])
return x;
return pre[x]=find(pre[x]);
}
bool join(int x,int y)
{
int fx=find(x);
int fy=find(y);
if(fx!=fy)
{
pre[fx]=fy;
return 1;
}
return 0;
}
int main()
{
int i,m;
while(cin>>n&&n)
{
int m=n*(n-1)/2;
for(i=1; i<=m; i++)
{
pre[i]=i;
}
for(i=1; i<=m; i++)
{
cin>>arr[i].u>>arr[i].v>>arr[i].w;
}
sort(arr+1,arr+m+1,cmp);
int sum=0;
for(i=1; i<=m; i++)
{
if(join(arr[i].u,arr[i].v))
{
sum+=arr[i].w;
}
}
cout<<sum<<endl;
}
return 0;
}