prim算法

Posted 2020-09-14 ewitt

#include <iostream>
#include <algorithm>
using namespace std;
#define MAX 5
#define INF 32765

int AdjMtx[MAX][MAX]=
{{INF ,1,INF,4,3},
{1,INF,2,INF,INF},
{INF,2,INF,2,3},
{4,INF,2,INF,1},
{3,INF,3,1,INF}};

int lowcost[5][2];//lowcost[][0]表示是否被选中，lowcost[][1]表示当前每个结点到选中结点的最小权值
int adjecent[MAX];
void prim(int st)
{
	int i;
	for(i=0;i<MAX;i++)
	{lowcost[i][0]=-1;adjecent[i]=st;}//初始化时每个节点的邻接点均设为st出发点
	for(i=0;i<MAX;i++)
		lowcost[i][1]=AdjMtx[st][i];
	lowcost[st][0]=0;//表示已经被选中
	
	int v=1;//控制结点加入个数
	
	while(v<MAX)
	{
		int tempV,preV;//tempV存放需要加入的结点 ,tempV的前驱是preV
		int min=INF;
		for (i=0;i<MAX;i++)
		{
			if(lowcost[i][0]==-1&&lowcost[i][1]<min)
			{
				min=lowcost[i][1];
				tempV=i;
			}		  
		}
		st=tempV;
		lowcost[st][0]=0;
		cout<<adjecent[st]<<"---->"<<st<<"edge added!"<<endl;
		for (i=0;i<MAX;i++)
		{
			if(lowcost[i][0]==-1&&AdjMtx[st][i]<lowcost[i][1])
			{lowcost[i][1]=AdjMtx[st][i];adjecent[i]=st;}
		}
		v++;
	}
}
void main()
{
	prim(3);
}