1004 Problem D

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;

}