最小生成树(C语言, prim算法)

图（来源：<<大话数据结构>>p250）

#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>
/*
* 邻接矩阵, prim普里姆算法(属贪婪算法)，无向图，最小生成树
* 代码实现<<大话数据结构>>p250 图7-6-6,v0至v8分别用ABCDEFGHI代替（不过打印过程还是用的下标）
* 最终成生n-1条边的树，路径权值和最小
*/

#define MAX 9
#define INFINITY 65535

// 图结构体
typedef struct {
char vexs[MAX]; // 顶点的数组，顶点类型为了简单使用char
int arc[MAX][MAX]; // 边表二维数组，值为权
int numVex;
}GRAPH, *PGRAPH;

void create(PGRAPH);
void gprint(GRAPH);
void prim(GRAPH);

void prim(GRAPH graph)
{
int i, j, k, min;

// 保存相关节点的数组（也可叫作父子(前后)关系，下标为当前节点，值为前一个节点，形成1条边）
int adjVex[MAX];
// 保存节点相关的边的最小权值(这个是随着程序不断迭代而更新的)
int lowcost[MAX];
// 循环处理前的初始化工作
adjVex[0] = 0; // 以第1个顶点为开头，直接加入v0节点
lowcost[0] = 0; // v0节点不需要再计算权值，标识为0，0还有个意思表示该节点已经加入最小生成树
// 使用v0节点相关的数据，初始化上面2个数组
for (i=0; i<graph.numVex; i++) {
//先全部初始化为0，表示所有节点的前1个节点都先为v0
adjVex[i] = 0;
// v0节点相关的边权值加入数组，因为入口是v0节点，这些是目前可以看到的相关的边
lowcost[i] = graph.arc[0][i];
}

/*
* 开始循环处理，次数为n-1，n为节点数
*/
// v0入口节点已经加入过数组不需要处理，所以从1开始
for (i=1; i<graph.numVex; i++) {
// 每轮都需要计算当前未加入最小生成树中的节点相关的最小权的边
int min = INFINITY;

// 先在lowcost数组中找出当前可以看到的边中，权值最小的那条边
for (j=1; j<graph.numVex; j++) {
if (lowcost[j] !=0 && lowcost[j] < min) {
min = lowcost[j];
k = j;
}
}

// 新找到的最小权值的边的相关节点为新查找根节点，标识为0，放入最小生成树
lowcost[k] = 0;
printf("%d->%d\n", adjVex[k], k); //adjVex可以知道相关节点前后关系

// 把符合条件的与新根节点(行)有关的边、节点信息更新到数组，供下一轮查找
for (j=1; j<graph.numVex; j++) {
if (lowcost[j] != 0 && graph.arc[k][j] < lowcost[j]) {
lowcost[j] = graph.arc[k][j];
adjVex[j] = k; // 只要找到的更新其前节点为k;
}
}
}
}

void create(PGRAPH g)
{
int i, j;

g->numVex = 9;
// 创建顶点
g->vexs[0] = 'A';
g->vexs[1] = 'B';
g->vexs[2] = 'C';
g->vexs[3] = 'D';
g->vexs[4] = 'E';
g->vexs[5] = 'F';
g->vexs[6] = 'G';
g->vexs[7] = 'H';
g->vexs[8] = 'I';
// 初始化边表
for (i=0; i<g->numVex; i++) {
for (j=0; j<g->numVex; j++) {
g->arc[i][j] = INFINITY;
if (j == i)
g->arc[i][j] = 0; //行列相等时表示自身，标识为0
}
}
// 添加边及权值
// A v0, B v1, C v2, D v3, E v4, F v5, G v6, H v7, I, v8
g->arc[0][1] = 10;
g->arc[1][0] = 10;
g->arc[0][5] = 11;
g->arc[5][0] = 11;
g->arc[1][2] = 18;
g->arc[2][1] = 18;
g->arc[1][8] = 12;
g->arc[8][1] = 12;
g->arc[1][6] = 16;
g->arc[6][1] = 16;
g->arc[2][8] = 8;
g->arc[8][2] = 8;
g->arc[2][3] = 22;
g->arc[3][2] = 22;
g->arc[3][8] = 21;
g->arc[8][3] = 21;
g->arc[3][6] = 24;
g->arc[6][3] = 24;
g->arc[3][7] = 16;
g->arc[7][3] = 16;
g->arc[3][4] = 20;
g->arc[4][3] = 20;
g->arc[4][7] = 7;
g->arc[7][4] = 7;
g->arc[4][5] = 26;
g->arc[5][4] = 26;
g->arc[5][6] = 17;
g->arc[6][5] = 17;
g->arc[6][7] = 19;
g->arc[7][6] = 19;
}

void gprint(GRAPH graph)
{
int i, j;
for (i=0; i<graph.numVex; i++) {
for (j=0; j<graph.numVex; j++){
printf("%6d ", graph.arc[i][j]);
}
putchar('\n');
}
}

int main(void)
{
GRAPH graph;
create(&graph);
gprint(graph);
prim(graph);
return 0;
}

output

[root@8be225462e66 c]# gcc prim.c && ./a.out
0     10  65535  65535  65535     11  65535  65535  65535
10      0     18  65535  65535  65535     16  65535     12
65535     18      0     22  65535  65535  65535  65535      8
65535  65535     22      0     20  65535     24     16     21
65535  65535  65535     20      0     26  65535      7  65535
11  65535  65535  65535     26      0     17  65535  65535
65535     16  65535     24  65535     17      0     19  65535
65535  65535  65535     16      7  65535     19      0  65535
65535     12      8     21  65535  65535  65535  65535      0
0->1
0->5
1->8
8->2
1->6
6->7
7->4
7->3
[root@8be225462e66 c]#