Description
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。
当N为0时,输入结束,该用例不被处理。
Output
对每个测试用例,在1行里输出最小的公路总长度。
可以当做模板学习
***************************************************************************
#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;
const int maxn = 105;
const int oo = 0x7fffffff;
int G[maxn][maxn];
int Prim(int N)
{
int dist[maxn], vis[maxn]={0, 1};
int ans=0, i, T = N-1;
for(i=1; i<=N; i++)
dist[i] = G[1][i];
while(T--)
{
int k=1, mini = oo;
for(i=1; i<=N; i++)
{
if(!vis[i] && mini > dist[i])
mini = dist[i], k=i;
}
vis[k] = true; ans += mini;
for(i=1; i<=N; i++)
if(!vis[i])dist[i] = min(dist[i], G[k][i]);
}
return ans;
}
int main()
{
int N;
while(scanf("%d", &N), N)
{
int M=(N-1)*N/2, u, v, w;
for(int i=1; i<=N; i++)
for(int j=1; j<=N; j++)
G[i][j] = (i==j? 0 : oo);
while(M--)
{
scanf("%d%d%d", &u, &v, &w);
G[u][v] = G[v][u] = min(G[u][v], w);
}
int ans = Prim(N);
printf("%d\n", ans);
}
return 0;
}