Prim是寻找最小生成树的算法,本文利用邻接矩阵,一个比较方便的方式来进行Prim算法。
邻接矩阵是存储有向或无向图的方式,若不带权,一般用 0 或 inf 表示未连接,用 1 表示连接。
如果带权,可以用一个矩阵来表示,-1是未连接,若连接,则数值等于权值;也可以用两个矩阵分别表示连接情况和权值。
本文使用一个矩阵,实现带权无向图的Prim算法。
#define inf 9999999
//M是无向图顶点个数,matrix是无向图的邻接矩阵
void Prim_tree(double** matrix, int M)
{
int i, j, start, Index = 0, x = 0, y = 0;
double min = inf;
int* T;
double** matrix2;
/*建立记录已经是结点的中间变量,生成树上有哪些节点,就向T中放*/
T = (int*)malloc(M * sizeof(int));
/*
*建立临时矩阵
*防止原来数据被改变
*/
matrix2 = (double **)malloc(M * sizeof(double));
for (i = 0; i < M; i++)
matrix2[i] = (double *)malloc(M * sizeof(double));
for (i = 0; i < M; i++)
for (j = 0; j < M; j++)
matrix2[i][j] = matrix[i][j];
/*输入开始的结点*/
printf("输入Prim算法开始的节点:\n");
printf("--> ");
scanf("%d", &start);
while (start > M || start < 1)
{
printf("--> ERROR!请重新输入:\n");
printf("--> ");
scanf("%d", &start);
}
printf("\n输入成功!\n");
/*开始记录和计算*/
T[Index] = start - 1;
while (Index < M-1)
{
min = inf;
for (i = 0; i <= Index; i++)
{
for (j = 0; j < M; j++)
{
if (matrix2[T[i]][j] != -1 && matrix2[T[i]][j] <= min)
{
x = T[i];
y = j;
min = matrix2[T[i]][j];
}
}
}
T[Index + 1] = y;
/*输出连接点*/
printf("\n连接 %d %d 权值 %0.3lf \n", x + 1, y + 1, min);
/*将已经连接的设为-1,防止重新计算或者形成回路*/
for (i = 0; i <= Index; i++)
{
matrix2[y][T[i]] = -1;
matrix2[T[i]][y] = -1;
}
Index++;
}
}