1399 最小生成树（Prim2）

1399 最小生成树

• 描述： 给出含n个顶点的图，打印最小生成树的所有边

• 输入：输入的第一行包含一个正整数n，表示图中共有n个顶点。其中n不超过50。
  以后的n行中每行有n个用空格隔开的整数，对于第i行的第j个整数，如果不为0，则表示第i个顶点和第j个顶点有直接连接且代价为相应的值，0表示没有直接连接。当i和j相等的时候，保证对应的整数为0。
  输入保证邻接矩阵为对称矩阵，即输入的图一定是无向图，且保证图中只有一个连通分量。

• 输出：每一行输出一条边

• 输入示例

  4
  0 2 4 0
  2 0 3 5
  4 3 0 1
  0 5 1 0
  

• 输出示例

  0->1
  1->2
  2->3
  

• 代码块

  #include<bits/stdc++.h>
  using namespace std;
  
  #define inf 99999
  
  const int N = 101;
  
  struct MGraph{
      int vexnum;
      int arcs[N][N]; 
  };
  
  void Prim(MGraph G){
      int n = G.vexnum;
      vector<int> lowcost(n);//保存相关点的权值
      vector<int> adjvex(n);//巨鹿相关顶点的下标
      lowcost[0] = 0;//初始化第一个权值为0；即v0加入生成树，lowcost的值为0，在这里就是此下标的顶点已经加入生成树
      for(int i=0; i<n; i++){
          lowcost[i] = G.arcs[0][i];//将v0顶点与之有边的权值存入数组
          adjvex[i] = 0;//初始化都为v0的下标
      } 
      for(int i=1; i<n; i++){//同样，这里的i表示要循环n-1次才能将所有的点加入生成树
          int k=0, min=inf;//初始化最小权值为无穷
          for(int j=1; j<n; j++){//循环所有顶点
              if(lowcost[j]!=0 && lowcost[j]<min){//这里不能写!lowcost[i],只能写lowcost[i]!=0
                  //如果当前权值成为最小值
                  min = lowcost[j];
                  k = j;
              }
          }
          printf("%d->%d\n", adjvex[k], k);//打印当前顶点边中权值最小边
          lowcost[k] = 0;//当前结点的权值设置为0， 表示此顶点已经完成任务
          for(int i=1; i<n; i++){//循环所有点
              if(lowcost[i]!=0 && lowcost[i]>G.arcs[k][i]){//这里不能写!lowcost[i],只能写lowcost[i]!=0
                  //若下标为k顶点各边权值小于此前这些顶点未被加入生成树权值
                  lowcost[i] = G.arcs[k][i];//将较小权值存入lowcost
                  adjvex[i] = k;//将下标为的顶点存入adjvex
              } 
          }
      }
  }
  
  int main(){
      int n;
      MGraph G;
      scanf("%d", &n);
      G.vexnum = n;
      for(int i=0; i<n; i++){
          for(int j=0; j<n; j++){
              scanf(" %d", &G.arcs[i][j]);
              if(i != j && G.arcs[i][j] == 0) G.arcs[i][j] = inf; 
          }
      }
      Prim(G);
      system("pause");
      return 0;
  }