邻接矩阵实现Prim算法

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