PTA 数据结构与算法 7-10 公路村村通

如有不对，不吝赐教
进入正题：
现有村落间道路的统计数据表中，列出了有可能建设成标准公路的若干条道路的成本，求使每个村落都有公路连通所需要的最低成本。

输入格式:
输入数据包括城镇数目正整数N（≤1000）和候选道路数目M（≤3N）；随后的M行对应M条道路，每行给出3个正整数，分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见，城镇从1到N编号。

输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通，则输出−1，表示需要建设更多公路。

输入样例:
6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3

输出样例:
12

这道题就是属于最小生成树的算法，一般有Kruskal算法和Prim算法，也还有Sollin算法，Boruvka算法等，可以去了解下,我使用Prim算法

Prim算法核心：

	for(j=1;j<N+1;j++){
        if(j==i)
          continue;          //费用小且不在生成树中
        if(costToTree[j]<min&&costToTree[j]){
          index=j;
          min=costToTree[j];                 //找到距当前节点且没有入树的最短节点
        }
      }
      if(-1==index)
        break;
      else{
        cost+=min;
        i=index;                //转移到这个节点 开始扩张
        count++;
        costToTree[i]=0;
      }
    }

关键就是维护一个各个节点到当前生成树的距离，每次选取最短的那个节点以及连接的边给加进来。

下面上代码 ：

#include<stdio.h>

#define INF 1000000

int main(void)
{
    int N,M;
    scanf("%d %d",&N,&M);
    int money[N+1][N+1];
    int i,j,count;
    int cost;
    int costToTree[N+1];        //距离树的距离

    for(i=1;i<N+1;i++){
      costToTree[i]=INF;
      for(j=1;j<N+1;j++)
        if(i==j)
          money[i][j]=money[j][i]=0;
        else
          money[i][j]=money[j][i]=INF;      //初始化
    }

    for(i=0;i<M;i++){
      int l1,l2,l3;
      scanf("%d %d %d",&l1,&l2,&l3);
      money[l1][l2]=money[l2][l1]=l3;
    }                                       //读入数据

    i=1;
    count=1;               //记录当前生成树的节点个数
    cost=0;                //花费

    while(count<N){
      for(j=1;j<N+1;j++)
        if(money[i][j]<costToTree[j])
          costToTree[j]=money[i][j];     //更新各节点到生成树的距离

      int min=INF;
      int index=-1;
      for(j=1;j<N+1;j++){
        if(j==i)
          continue;          //费用小且不在生成树中
        if(costToTree[j]<min&&costToTree[j]){
          index=j;
          min=costToTree[j];                 //找到距当前节点且没有入树的最短节点
        }
      }
      if(-1==index)
        break;
      else{
        cost+=min;
        i=index;                //转移到这个节点 开始扩张
        count++;
        costToTree[i]=0;
      }
    }

    if(count==N)
      printf("%d",cost);
    else
      printf("-1");

    return 0;
}

结果（注：测试的时候忘改编译器了，不过没啥影响）：