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;
}
程序运行结果: