Prim算法基本思想
设G=(V,E)是无向连通网,T=(U,TE)是G的最小生成树,Prim算法的基本思想是:从初始状态U={v}(v∈V)、TE={ }开始,重复执行下述操作:在所有i∈U,j∈V-U的边中找一条代价小的边(i,j)并人集合TE,同时j并入U,直至U=V为止,此时TE中有n-1条边,T是一棵最小生成树。
简单来说,Prim算法就是在不产生回路的前提下,从任意结点开始,作为最小生成树的第一个结点,然后依次查找与已加入最小生成树的结点相连的最短边的另一结点加入到最小生成树中,直到所有结点加入。
伪代码:
算法:Prim
输人:无向连通网G=(V,E)
输出:最小生成树T=(U,TE)
1.初始化:U={v};TE={};
2.重复下述操作直到U=V:
2.1 在E中寻找最短边(i,j),且满足i∈U,j∈V-U;
2.2 U=U+{j};
2.3 TE=TE+{(i,j)};
Prim算法基于的存储结构
图的存储:在算法执行过程中,需要不断的读取任意两顶点之间的权值,所以采用邻接矩阵存储。
候选最短边集:设置两个数组分别为adjvex[n],lowcost[n]。
adjvex表示候选最短边的临接点,lowcost表示候选最短边的权值。规定lowcost[v]=0表示该点加入了最小生成树。
如果候选最短边为(i,j),那么
adjvex[i]=j;//j为邻接点
lowcost[i]=w//w为最短边权值
初始时将lowcost[v]=0,表示选取v作为最小生成树的第一个结点。adjvex[i]=v,lowcost[i]=边(v,i)的权值。
在每一次迭代中,首先在数组lowcost[n]中选取权值最小的lowcost[j],然后更新adjvex[n]与lowcost[n],再将lowcost[j]置为0,即将j加入最小生成树中。下为过程图。
Prim实现最小生成树代码
输出结果为选取最短边的权值,如果要输出结点,自行在构造函数中vertex[]数组存储结点,
在Prim中输出权值时输出相应结点即可。
#include<iostream>
using namespace std;
const int maxnum=10;
int Minedge(int a[],int n) //找出最小权值的边
{
int min=101;
int x=0;
for(int i=1;i<n;i++)
{
if(a[i]<min&&a[i]!=0){
min=a[i];
x=i;
}
}
return x;
}
class MGraph
{
public:
MGraph(int a[maxnum][maxnum],int n,int e);
void Prim(int v);
private:
int edge[maxnum][maxnum];
int vertexnum;
int edgenum;
};
MGraph::MGraph(int a[maxnum][maxnum],int n,int e)
{
int i,j;
vertexnum=n;
edgenum=e;
for(i=0;i<n;i++){
for(j=0;j<n;j++){
edge[i][j]=a[i][j];
}
}
}
void MGraph::Prim(int v)
{
int i,j,k;
int adjvex[maxnum],lowcost[maxnum];
for(i=0;i<vertexnum;i++)
{
lowcost[i]=edge[v][i];
adjvex[i]=v;
}
lowcost[v]=0;
for(k=1;k<vertexnum;k++)
{
j=Minedge(lowcost,vertexnum);
cout<<lowcost[j]<<" ";
lowcost[j]=0;
for(i=0;i<vertexnum;i++){
if(edge[i][j]<lowcost[i]){
lowcost[i]=edge[i][j];
adjvex[i]=j;
}
}
}
}
int main()
{
int m,n;
int a[maxnum][maxnum];
cin>>m>>n;
for(int i=0;i<m;i++){
for(int j=0;j<m;j++){
cin>>a[i][j];
}
}
MGraph mg(a,m,n);
mg.Prim(0);
return 0;
}