纯C语言：贪心Prim算法生成树问题源码分享

复制代码 代码如下:

#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;
}