1004 Problem D
题意:某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。计算最小的公路总长度。
思路:求连通所有村庄的最小公路总长度,即求最小生成树问题,利用prim算法,将村庄看做一个图,求最小生成树。
感想:prim算法本身并不难理解,不过可能在实现的过程中,多少有些困难,还是做题太少。
#include<iostream>
#include<string.h>
#include<algorithm>
const int N=101;
int map[N][N];
int r[N];
int i,j,n;
int prim(){
int sum=0;
int k,min,t=n;
while(--t){
min=100000;
for(i=2;i<=n;i++){
if(r[i]!=1&&min>map[1][i]){
min=map[1][i];
k=i;
}
}
sum+=min;
r[k]=1;
for(i=2;i<=n;i++){
if(r[i]!=1&&map[k][i]<map[1][i]){
map[1][i]=map[k][i];
}
}
}
return sum;
}
int main(){
int a,b,d,m;
while(scanf("%d",&n)&&n!=0){
m=n*(n-1)/2;
memset(map,0,sizeof(map));
memset(r,0,sizeof(r));
for(i=1;i<=m;i++){
scanf("%d%d%d",&a,&b,&d);
map[a][b]=map[b][a]=d;
}
printf("%d\n",prim());
}
return 0;
}