不同于克鲁斯卡尔生成树,每次寻找权值最小的边进行连接,普利姆生成树的规则是,选取1节点为树根,每次寻找最靠近生成树的节点加入生成树,并修改树外节点到树的距离。
普利姆适应于点少的情况,当题目给出点坐标而非边权值时,往往利用克鲁斯卡尔会陷入对大量边的计算,这时普利姆的优势得以体现。
细节如下
一,初始化dis数组为inf
二,dis[i]数组赋初值时,赋成w[1][i],表示到根节点距离
三,标记数组事先标记1节点
四,每次找寻树外节点(未被标记),找到之后别忘了标记,并且记录答案
五,借助这一节点去更新树外节点dis值
# include<iostream>
# include<math.h>
# define inf 0x7f7f7f7f
using namespace std;
typedef long long int ll;
int w[1000][1000];
bool book[1000];
int dis[1000];
int n;
int main ()
{
fill(dis,dis+1000,99999);
cin>>n;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
cin>>w[i][j];
}
}
for(int i=1;i<=n;i++)
{
dis[i]=w[1][i];
}
dis[1]=0;
book[1]=1;
int now,min1,sum=0;
for(int i=1;i<n;i++) //找寻n-1个点
{
min1=inf;
for(int j=1;j<=n;j++)
{
if(book[j]==0)
{
if(dis[j]<min1)
{
min1=dis[j];
now=j;
}
}
}
sum+=min1;
book[now]=1;
for(int j=1;j<=n;j++)
{
if(book[j]==0)
{
if(dis[j]>w[now][j])
{
dis[j]=w[now][j];
}
}
}
}
return 0;
}