题目描述:
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
输入:
测试输入包含若干测试用例。每个测试用例的第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
来源:
2006年浙江大学计算机及软件工程研究生机试真题
用Prime算法(C语言)
#include<stdio.h>
#include<string.h>
#define N 100
#define MAX 10000
int prim(int path[N][N], int n){
int sum= 0, min= MAX;//总路径, 目前最短
int pa, pb;//最短端点
int visited[N];
int num;//走过的点数
int i, j;
memset(visited, 0, sizeof(visited));
visited[1]= 1;//1点走过
num= 1;
while(num< n){
min= MAX;
for(i= 1; i<= n; i++){
if(visited[i]== 1){
for(j= 1; j<= n; j++){
if(visited[j]== 0 && path[i][j]> 0){
if(path[i][j]< min){
min= path[i][j];
pa= i;
pb= j;
}
}
}
}
}
visited[pb]= 1;
sum+= path[pa][pb];
num++;
}
return sum;
}
int main(){
int n;//村庄数
int path[N][N];//路径长度
int a, b, c;//临时变量
while(scanf("%d", &n)!=EOF && n!= 0){
memset(path, 0, sizeof(path));
for(int i= 0; i< n*(n-1)/2; i++){
scanf("%d %d %d", &a, &b, &c);
path[a][b]= path[b][a]= c;
}
printf("%d\n", prim(path, n));
}
return 0;
}