数据结构（C++）Prime算法

Prime和Kruskal两种算法是求图最小生成树的两种算法，Prime是找点，Kruskal是找边，各有各的有点，也其实都有点小难。
最小生成树的点数等于原先图的点数，边数等于点数-1 。

使用邻接矩阵，关于图的数组存储之前就已经写过了点击这里

图数组存储的构造函数

依然保持着自己的习惯，每个顶点都从1开始计算，不从0，方便输入。
比之前的构造函数增加了边权，如果没有赋权则为INF（无穷大）。

MGraph(int a[], int n, int e)
	{
		arcNum = e;
		vertexNum = n;
		for (int i = 1; i <= vertexNum; ++i)
		{
			vertex[i] = a[i];
		}
		for (int i = 1; i <= vertexNum; ++i)
		{
			for (int j = 1; j <= vertexNum; ++j)
			{
				arc[i][j] = INF;
			}
		}
		for (int i = 1; i <= arcNum; ++i)
		{
			int a, b, v;
			cin >> a >> b >> v;
			arc[a][b] = arc[b][a] = v;
		}
	}

Prime算法思想

我们把一个整图定义为V，顶点（第一个开始的点）所在的集合成为U，这样还没有连接到最小生成树中的顶点就属于集合（U-V）中。

这里还需要定义两个额外的辅助数组：

• lowcost[n]：用来表示（V-U）中的各个顶点与集合（U）中的顶点最短边的权值，如果lowcost[v]=0，则表示顶点 v 已经加入到最小生成树之中。
• adjvex[n]：用来保存集合（U-V）中的顶点与集合（U）中连接的最小边的顶点。这个顶点是 U 中的。

代码思想：

1. 首先初始化 两个辅助数组，lowcost[i]=arc[1][i] ，adjdex[i]=1 。 表示以1为最开始的顶点，与1顶点相连接的边的边权就等于arc数组的值。 adjdex为1表示图V中各个顶点的连接都与1顶点相连接。
2. 给lowcost[1]=0，设置起点顶点到自身距离为0 。
3. 执行n-1次循环。
   3.1. 在lowcost[n]中选取最小边，lowcost[k]，去找到相应的顶点k。
   3.2. 输出顶点k的顶点信息和与起点顶点对应的权值。
   3.3. 将顶点k加入的集合U之中，（lowcost[k]=0）
   3.4. 调整lowcost和adjvex数组

调整lowcost和adjvex数组

由于lowcost数组表示的是U-V中顶点与U中顶点相连的最小边，当加入了一个顶点后，所有的lowcost和adjvex都有可能发生相应的变化。
就是一个很简单的判断，刚开始一下子蒙住了，真是不应该。

for (int j = 1; j <= vertexNum; ++j)
{
	if (arc[k][j] < lowcost[j])
	{
		lowcost[j] = arc[k][j];
		adjvex[j] = k;
	}
}

Prime算法实现

void Prime()
{
	for (int i = 2; i <= vertexNum; ++i)
	{
		lowcost[i] = arc[1][i];
		adjvex[i] = 1;
	}
	lowcost[1] = 0;
	for (int i = 2; i <= vertexNum; ++i)
	{
		int k = MinEdge(lowcost, vertexNum);
		cout << k << "  " << adjvex[k] << "  " << lowcost[k] << endl;
		lowcost[k] = 0;
		for (int j = 1; j <= vertexNum; ++j)
		{
			if (arc[k][j] < lowcost[j])
			{
				lowcost[j] = arc[k][j];
				adjvex[j] = k;
			}
		}
	}
}

全部代码

• visited数组没有用到，如果要进行遍历时需要。
• MinEdge是用来算出lowcost数组中最小值的 i 点，也就是顶点，第一次搞错了，一位是直接返回lowcost值，其实不是，应该返回顶点值。这里还需要注意的是值为0时一定是最小的，但是不能计算进去，因为这个时候为0的点已经是集合U中的了，这里需要额外判断一步。
• 图论的算法每次都想从1开始，改代码改成憨憨了。

#include<iostream>
using namespace std;
const int MaxSize = 20;
const int INF = 2147483647;
int visited[MaxSize];
int lowcost[MaxSize];
int adjvex[MaxSize];
int MinEdge(int a[], int n)
{
	int min = INF;
	int k = 0;
	for (int i = 1; i <= n; ++i)
	{
		if (min > a[i] && a[i] != 0)
		{
			min = a[i];
			k = i;
		}
	}
	return k;
}
class MGraph
{
private:
	int arc[MaxSize][MaxSize];
	int vertex[MaxSize];
	int arcNum;
	int vertexNum;
public:
	MGraph(int a[], int n, int e)
	{
		arcNum = e;
		vertexNum = n;
		for (int i = 1; i <= vertexNum; ++i)
		{
			vertex[i] = a[i];
		}
		for (int i = 1; i <= vertexNum; ++i)
		{
			for (int j = 1; j <= vertexNum; ++j)
			{
				arc[i][j] = INF;
			}
		}
		for (int i = 1; i <= arcNum; ++i)
		{
			int a, b, v;
			cin >> a >> b >> v;
			arc[a][b] = arc[b][a] = v;
		}
	}
	void Prime()
	{
		for (int i = 2; i <= vertexNum; ++i)
		{
			lowcost[i] = arc[1][i];
			adjvex[i] = 1;
		}
		lowcost[1] = 0;
		for (int i = 2; i <= vertexNum; ++i)
		{
			int k = MinEdge(lowcost, vertexNum);
			cout << k << "  " << adjvex[k] << "  " << lowcost[k] << endl;
			lowcost[k] = 0;
			for (int j = 1; j <= vertexNum; ++j)
			{
				if (arc[k][j] < lowcost[j])
				{
					lowcost[j] = arc[k][j];
					adjvex[j] = k;
				}
			}
		}
	}
};
int main()
{
	int a[MaxSize];
	for (int i = 1; i < MaxSize; ++i)
	{
		a[i] = i;
	}
	int x, y;
	cin >> x >> y;
	MGraph G(a, x, y);
	G.Prime();
	return 0;
}

测试用例

6 9
1 2 34
1 3 46
1 6 19
3 4 17
3 6 25
4 6 25
4 5 38
2 5 12
5 6 26

我们都要越来越优秀