prime算法简单来讲就是将一群点分两个集合,S集合代表已经选入生成树的点,U集合代表未选入的点。
算法过程就是将U集合中距离S集合最短的点加入S集合直至U集合元素为0;
(图片来源:https://www.jianshu.com/p/ff6db00ad866)
代码实现
#include<iostream>
using namespace std;
#define MAX 10000
int map[5][5]={
{0,4,MAX,2,MAX},
{4,0,4,1,MAX},
{MAX,4,0,1,3},
{2,1,1,0,7},
{MAX,MAX,3,7,0}
};
int main()
{
int count=0;
int S[5]={1,0,0,0,0},U[5]={0,1,1,1,1};
for(int k=0;k<4;k++)
{
int dim=MAX,temp1=0,temp2=0;
for(int i=0;i<5;i++)
{
if(S[i])
{
for(int j=0;j<5;j++)
{
if((U[j]==1)&&(map[j][i]<dim))
{
dim=map[i][j];
temp1=j;
temp2=i;
}
}
}
}
S[temp1]=1;U[temp1]=0;
cout<<(char)('A'+temp2)<<(char)('A'+temp1)<<endl;
}
return 0;
}
运行截图
由于本人能力有限,算法的时间复杂度还是很高的O(n3),如有改进定会修改