最小生成树 prim算法

 

   https://blog.csdn.net/hnust_xiehonghao/article/details/38013125帮助理解算法原理

输入：无向带权图 G=(v,e),对于每一条边都有权值 e,

输

出：这个图的最小生成树，即一个连接所有点的树，且权值最小，没有回路

设 G=（V,E）是连通带权图，U是V 的真子集。如果（v1,e1）属于E，且v1 属于U，e1 属于 V- U ，且在所有的边里面（v1， e1）权值最小，那么一定存在G的最小生成树

prim算法的思路：

   这个算法是以图的顶点为基础，从一个初始顶点开始，寻找触达其他顶点权值最小的边，并把该顶点加入到已触达顶点的集合中。当全部顶点都加入到集合时，算法的工作就完成了。Prim算法的本质，是基于贪心算法。

接下来说一说最小生成树的存储方式。我们最常见的树的存储方式，是链式存储，每一个节点包含若干孩子节点的指针，每一个孩子节点又包含更多孩子节点的指针：

这是一个连通图

  

   我们顺着算法走一遍

    找到初始顶点，加入已达顶点集中       

 

 

在已达顶点集中，寻找新的顶点且权值最小的，0到2的权值最小，在parent中记录2的父节点parent[2]=0;

  

在0，2 已达顶点集中显然，2到4的权值最小，顶点4的父子节点为2所以parents[4]=2;

 

在已达顶点集中，只有0，2，4可供我们选择，显然0到1的权值最小，顶点1的父子节点为0所以parent[1]=0

在已达顶点集到顶点3的只有1，4，显然3到1的权值更小，parent[3]=1;

这样下来所有的点都会触达的，最小生成树存储在parent 数组中

上代码

#include <iostream>
#include <vector>
using namespace std;
vector<int>a;
int parent[50];
const int INF= 100000;
int matrix[50][50]={
	{0,4,3,INF,INF},
            {4,0,8,7,INF},
            {3,8,0,INF,1},
            {INF,7,INF,0,9},
            {INF,INF,1,9,0},
};

int main()
{
int n;
cin>>n;
int weight;
int toindex=0;
int formindex=0;
a.push_back(0);
parent[0]=-1;
while(a.size()<n){
weight=INF;
//在已到达的顶点里面寻找权值最小的
for(int i=0;i<a.size();i++){
	 for(int j=0;j<n;j++){
		 int flag=0;
		 for(int t=0;t<a.size();t++){
			 if(j==a[t])//查找是否为已到顶点
				 flag=1;
		 }
		 if(!flag&&matrix[i][j]){
			 if(matrix[i][j]<weight){
				 formindex=j;
				 toindex=i;
				 weight=matrix[i][j];
			 }
		 }
	 }

}
parent[formindex]=toindex;
a.push_back(formindex);
}
for(int i=0;i<n;i++)
   cout<<parent[i]<<" ";
	return 0;
}