还是畅通工程

还是畅通工程

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

Input测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。
Output对每个测试用例，在1行里输出最小的公路总长度。
Sample Input

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

Sample Output

3

5

#include <stdio.h>
#include <string.h>

#define INF 200000000
#define MAX 105

int Prim(int n, int map[][MAX])
{ //最小生成树算法
  
  //存放集合s中顶点以外的顶点到s的最短距离
  int lowCost[MAX];   
  //存放集合s中顶点以外的顶点到s的最短距离的顶点
  int closest[MAX];   
  int s[MAX];   //存放已生成的顶点
  int min;
  int u;
  int dist = 0;   //最小生成树的总长度
  int i, j;
  //初始化数组
  memset(s,0,sizeof(s));
  memset(lowCost, 0, sizeof(lowCost));
  memset(closest, 0, sizeof(closest));
  for (i=1; i<=n; i++)
  {   //以1为起点建立最小生成树
    lowCost[i] = map[1][i];
    closest[i] = 1;
  }
  s[1] = 1;   //将起点1放入集合s
  for (j=1; j<=n-1; j++)
  {   //进行n-1次循环
    min = INF;
    u = 1;
    //找到距i最近的顶点
    for (i=1; i<=n; i++)
      if (lowCost[i]<min&&!s[i])
      {
        min = lowCost[i];
        u = i;
      }
    s[u] = 1;
    dist += lowCost[u];
    //用顶点u跟新lowCost和closest
    for (i=1; i<=n; i++)
      if (map[u][i]<lowCost[i]&&!s[i])
      {
        lowCost[i] = map[u][i];
        closest[i] = u;
      }
  }
  return dist;
}

int main()
{
    int n;
    int map[MAX][MAX];  //邻接矩阵
    int a,b,c;
    int dist;
    int i;
    while (~scanf("%d",&n))
    {
      if (n==0)
        break;
      for (i=1; i<=n*(n-1)/2; i++)
      {
        scanf("%d%d%d",&a,&b,&c);
        map[a][b] = map[b][a] = c;
      }
      dist = Prim(n,map);
      printf("%d\n",dist);
    }
    return 0;
}