prim算法

prim算法是用来求加权连通图最小生成树算法，算法的执行步骤如下：

对于图G而言，V是所有顶点的集合；现在，设置两个新的集合S和T，其中S用于存放G的最小生成树中的顶点，T存放G的最小生成树中的边。 从所有uЄS，vЄ(V-S) (V-S表示出去S的所有顶点)的边中选取权值最小的边(u, v)，将顶点v加入集合S中，将边(u, v)加入集合T中，如此不断重复，直到S=V为止，最小生成树构造完毕，这时集合T中包含了最小生成树中的所有边，Prim算法利用贪心算法的思想来解决最小生成树，利用最小生成树的性质和数学归纳法可以很容易证明prim算法的正确性。

这里以下面这个图为例，实现prim算法，求出这个图最小生成树。

// ConsoleApplication3.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include<iostream>
using namespace std;

const int MAX = 7;
int edge[MAX][MAX];

void prim();
void test()
{
	for (int i = 0; i < MAX; i++)
	for (int j = 0; j < MAX; j++)
		edge[i][j] = INT_MAX;
	//手动构造一个如上图所示的图
	edge[0][1] = 12;
	edge[0][6] = 14;
	edge[0][5] = 16;

	edge[1][0] = 12;
	edge[1][2] = 10;
	edge[1][5] = 7;

	edge[2][1] = 10;
	edge[2][3] = 3;
	edge[2][4] = 5;

	edge[3][2] = 3;
	edge[3][4] = 4;

	edge[4][2] = 5;
	edge[4][3] = 4;
	edge[4][5] = 2;
	edge[4][6] = 8;

	edge[5][0] = 16;
	edge[5][1] = 7;
	edge[5][2] = 6;
	edge[5][4] = 2;
	edge[5][6] = 9;

	edge[6][0] = 14;
	edge[6][4] = 8;
	edge[6][5] = 9;
	prim();
}

void prim()
{
	bool visited[MAX];	//表示节点i是否已经添加到结果集合S中
	int lowestCost[MAX];	//记录集合S中点到达集合V-S中点的最小距离
	int closest[MAX];		//记录集合V-S中的点到S中哪个点最近
	for (int i = 0; i < MAX; i++)	//初始化visited为false
	{
		visited[i] = false;
		lowestCost[i] = edge[0][i];
		closest[i] = 0;	//初始阶段，S中只有节点0，故V-S 集合中节点都离节点0最近
		visited[0] = true;	//将节点0加入集合S中
	}
	cout << "添加顺序：" << endl << 0;
	for (int i = 1; i < MAX; i++)
	{
		int min = INT_MAX;
		int k = 0;	
		for (int j = 0; j < MAX; j++)
		{
			if (lowestCost[j] < min && !visited[j])
			{
				min = lowestCost[j];	//将距离S中节点最近的点加入到S中
				k = j;
			}
		}

		cout<<" "<< k ;

		visited[k] = true;
		for (int j = 0; j < MAX; j++)
		{
			//在将节点k加入s集合后，更新lowestCost数组
			if (edge[k][j] < lowestCost[j] && !visited[j])
			{
				lowestCost[j] = edge[k][j];	
				closest[j] = k;
			}
		}
	}
	
	cout <<endl<< "edge" << "  cost" << endl;
	for (int i = 1; i < MAX; i++)
	{
		cout << i << "--" << closest[i] << "  " << edge[i][closest[i]] << endl;
	}
	cout << endl;
}

int _tmain(int argc, _TCHAR* argv[])
{
	test();
	return 0;
}

程序运行结果：