C语言实现最小生成树之普里姆算法

实现的文字说明

如何实现的图示说明

mst[2]=1 lowcost[2]=6 从点1到点2的权值是6

以v3为终点的权值最小，把边<MST[3],3>=1加入最小生成树MST

以v6为终点的权值最小，把边<MST[6],6>=4加入最小生成树MST

以v4为终点的权值最小，把边<MST[4],4>=2加入最小生成树MST

以v2为终点的权值最小，把边<MST[2],2>=5加入最小生成树MST

当所有的数组的值都为0，说明创建成功。

在邻接矩阵中实现MST

//获取两个顶点之间的权值
E  GetWeight(GraphMtx *g, int v1, int v2)
{
	//v1,v2都是顶点的位置
	if (v1 == -1 || v2 == -1)
		return MAX_COST;
	return g->Edge[v1][v2];
}

//最小生成树
void MinSpanTree_Prim(GraphMtx *g, T vertex)
{
	int n = g->NumVertices;
	E *lowcost = (E*)malloc(sizeof(E)*n); //lowcost[n]
	int *mst = (int *)malloc(sizeof(int)*n);//mst[n]
	assert(lowcost != NULL && mst != NULL);

	int k = GetVertexPos(g, vertex);
	for (int i = 0; i < n; ++i)
	{
		if (i != k)
		{
			lowcost[i] = GetWeight(g, k, i);
			mst[i] = k;
		}
		else
		{
			lowcost[i] = 0;
		}
	}

	int min, min_index,i,j;
	int begin, end;
	E cost;
	for (i = 0; i < n - 1; ++i)
	{
		min = MAX_COST;
		min_index = -1;
		for (int j = 0; j < n; ++j)
		{
			if (lowcost[j] != 0 && lowcost[j] < min)
			{
				min = lowcost[j];
				min_index = j;
			}
		}
		
		begin = mst[min_index];
		end = min_index;
		printf("%c-->%c : %d\n", g->VerticesList[begin], g->VerticesList[end], min);

		lowcost[min_index] = 0;

		for (j = 0; j < n; ++j)
		{
			cost = GetWeight(g, min_index, j);
			if (cost < lowcost[j])
			{
				lowcost[j] = cost;
				mst[j] = min_index;
			}
		}
	}
}

分析

1.为两个数组分配空间，lowcost[i]存放某个顶点X到i的权值，mst[i]存放的是这个顶点X的位置

一开始权值初始化为无穷大，自己到对自己的权值为0

    int n = g->NumVertices;
	E *lowcost = (E*)malloc(sizeof(E)*n); //lowcost[n]  权值，每一次找一个顶点与其他顶点的距离，有几个顶点就需要存几个权值，
	int *mst = (int *)malloc(sizeof(int)*n);//mst[n]   顶点的位置
	assert(lowcost != NULL && mst != NULL);

2.初始化输入的起始顶点（假设是A）到各个顶点的权值lowcost[i]

	int k = GetVertexPos(g, vertex);//初始顶点的位置
	for (int i = 0; i < n; ++i)
	{
		if (i != k)
		{
			lowcost[i] = GetWeight(g, k, i);//起始顶点到各个顶点的权值存起来
			mst[i] = k;//这个起始顶点的位置就是k
			//也可以理解成对应于k到lowcost[i]中i的权值
		}
		else
		{
			lowcost[i] = 0;
		}
	}

3.找到A到各个顶点最小的权值——应该是lowcost[2],，和那个顶点所在的位置——2，顶点是C

		min = MAX_COST;//最小权值
		min_index = -1;//到最小权值的那个顶点的位置
		for (int j = 0; j < n; ++j)
		{
			if (lowcost[j] != 0 && lowcost[j] < min)
			{
				min = lowcost[j];
				min_index = j;
			}
		}

4.找到起点和终点，打印输出，并从c开始找C与各个顶点的权值，更新lowcost[i]，mst[i]的内容

第一次打印（即A-C，1），begin = mst[2]=0，end = min_index=2;

	begin = mst[min_index];
	end = min_index;
	printf("%c-->%c : %d\n", g->VerticesList[begin], g->VerticesList[end], min);

	lowcost[min_index] = 0;//说明这一次的一个顶点到另一个顶点的路径符合最小权值的要求，加入到最小生成树

	for (j = 0; j < n; ++j)
	{
		cost = GetWeight(g, min_index, j);//从c开始找C与各个顶点的权值
		if (cost < lowcost[j])//与原来存在的进行对比，若小于原来的就更新
		{
			lowcost[j] = cost;
			mst[j] = min_index;//这个顶点现在的C了
		}
	}

至此完成了第一个顶点A——C路径的选择，并更新了C与各个顶点的权值（原来是A到各个顶点）。若有n个顶点，则必有n-1条边才能生成最小生成树，所以要比较n-1次

for (i = 0; i < n - 1; ++i)
{
把3,4的步骤重新遍历n-1次
}