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

int graph[MAX][MAX];

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

 /* mst[i]記錄對應(yīng)lowcost[i]的起點(diǎn) */
 int mst[MAX];

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

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

  /* 標(biāo)記所有節(jié)點(diǎn)的起點(diǎn)皆為默認(rèn)的0號節(jié)點(diǎn) */
  mst[i] = 0;
 }

 /* 標(biāo)記0號節(jié)點(diǎn)加入生成樹 */
 lowcost[0] = 0;

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

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

  /* 標(biāo)記節(jié)點(diǎn)minid加入生成樹 */
  lowcost[minid] = 0;

  /* 更新當(dāng)前節(jié)點(diǎn)minid到其他節(jié)點(diǎn)的權(quán)值 */
  for (j = 1; j < n; j++)
  {
   /* 發(fā)現(xiàn)更小的權(quán)值 */
   if (graph[minid][j] < lowcost[j])
   {
    /* 更新權(quán)值信息 */
    lowcost[j] = graph[minid][j];

    /* 更新最小權(quán)值邊的起點(diǎn) */
    mst[j] = minid;
   }
  }
 }
 /* 返回最小權(quán)值和 */
 return sum;
}

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

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