贪心算法prim算法

普利姆算法(Prim):

算法思想基于顶点选择，通过维护两个点集，已选点集S和未选点集T，每次选择从S集映射到T集的最小代价边，适用于稠密图。

图解dis更新过程：

最短路径求解时，dis更新的都是以北京为起点，prim算法起点始终是从S集合出发向T集合连线，也就是dis表中找到权值最小的那个点，放入S集合中。

#include"stdafx.h"
#include<iostream>
#define INF 10000
using namespace std;
const int N = 6;
bool visit[N];
int dist[N] = { INF, };
int graph[N][N] = { { INF, 6, 1, 5, INF, INF },  //INF代表两点之间不可达
                    { 6, INF, 5, INF, 3, INF },
                    { 1, 5, INF, 5, 6, 4 },
                    { 5, INF, 5, INF, INF, 2 },
                    { INF, 3, 6, INF, INF, 6 },
                    { INF, INF, 4, 2, 6, INF }
				  };
int prim(int cur)
{
	int index = cur;
	int sum = 0;
	int i = 0;
	int j = 0;
	cout << index+1 << " ";
	memset(visit, false, sizeof(visit));
	visit[cur] = true;
	for (i = 0; i < N; i++)
		dist[i] = graph[cur][i];//初始化，每个与a邻接的点的距离存入dist
	for (i = 1; i < N; i++)
	{ 
		int minor = INF;
		for (j = 0; j < N; j++)
		{
			if (!visit[j] && dist[j] < minor)          //找到未访问的点中，距离当前最小生成树距离最小的点
			{
				minor = dist[j];
				index = j;
			}
		}
		visit[index] = true;
		cout << index+1 << " ";
		sum += minor;
		for (j = 0; j < N; j++)
		{
			if (!visit[j] && dist[j]>graph[index][j])      //执行更新，如果点距离当前点的距离更近，就更新dist
			{
				dist[j] = graph[index][j];
			}
		}
	}
	cout << endl;
	return sum;               //返回最小生成树的总路径值
}
int main()
{
	cout << prim(0) << endl;//从顶点a开始


	system("pause");
	return 0;
}