#include <iostream.h>
#define MAX 100
#define MAXCOST 100000

int graph[MAX][MAX];

int Prim(int graph[MAX][MAX], int n)
{
 /* lowcost[i]記錄以i為終點的邊的最小權值，當lowcost[i]=0時表示終點i加入生成樹 */
 int lowcost[MAX];

 /* mst[i]記錄對應lowcost[i]的起點 */
 int mst[MAX];

 int i, j, min, minid, sum = 0;

 /* 默認選擇0號節點加入生成樹，從1號節點開始初始化 */
 for (i = 1; i < n; i++)
 {
  /* 最短距離初始化為其他節點到0號節點的距離 */
   lowcost[i] = graph[0][i];

  /* 標記所有節點的起點皆為默認的0號節點 */
  mst[i] = 0;
 }

 /* 標記0號節點加入生成樹 */
 lowcost[0] = 0;

 /* n個節點至少需要n-1條邊構成最小生成樹 */
 for (i = 1; i < n; i++)
 {
  min = MAXCOST;
  minid = 0;

  /* 找滿足條件的最小權值邊的節點minid */
  for (j =1; j <n; j++)
  {
   /* 邊權值較小且不在生成樹中 */
   if (lowcost[j] < min && lowcost[j] != 0)
   {
    min = lowcost[j];
    minid = j;
   }
  }
  /* 輸出生成樹邊的信息:起點，終點，權值 */
  cout<<"生成數邊的起點、終點及權值分別為："<< mst[minid]+1<<"  "<<minid+1<<"  "<<min<<endl;
  /* 累加權值 */
  sum += min;

  /* 標記節點minid加入生成樹 */
  lowcost[minid] = 0;

  /* 更新當前節點minid到其他節點的權值 */
  for (j = 1; j < n; j++)
  {
   /* 發現更小的權值 */
   if (graph[minid][j] < lowcost[j])
   {
    /* 更新權值信息 */
    lowcost[j] = graph[minid][j];

    /* 更新最小權值邊的起點 */
    mst[j] = minid;
   }
  }
 }
 /* 返回最小權值和 */
 return sum;
}

void main()
{
 int i, j,  m,n;
 int  cost;
  /* 讀取節點的數目 */
 cout<<"請輸入該圖結點個數：";
 cin>>m;
 /* 初始化圖，所有節點間距離為無窮大 */
 for (i = 0; i <m; i++)
 {
  for (j =i+1; j <m; j++)
  {
   cout<<"請輸入結點"<<i+1<<"到結點"<<j+1<<"邊的權值，若無邊則輸入MAXCOST(100000):";
   cin>>n;
   graph[i][j] = n;
   graph[j][i] = n;
  }
  graph[i][i]=MAXCOST;
 }

 /* 求解最小生成樹 */
 cost = Prim(graph, m);
 cout<<"最小生成樹的權值為："<<cost<<endl;
}