代码手抄自《啊哈!算法》
#include<iostream>
using namespace std;
int main()
{
int n,m,i,j,k,min,t1,t2,t3;
int e[101][101],dis[101],book[101]={0};
int inf=999999;
int count=0,sum=0;//count用来记录生成树中顶点的个数,sum用来储存路路径之和
//读入n和m,n为顶点,m为边的数量
scanf("%d %d", &n, &m);
//初始化
for(i=1;i<=n;i++)
{
for(j=1;j<=n;j++)
{
if(i==j)
{
e[i][j]=0;
}
else
{
e[i][j]=inf;
}
}
}
//开始读入边
for(i=1;i<=m;i++)
{
scanf("%d %d %d",&t1,&t2,&t3);
//注意是无向图,所以需要将边再反向储存一边
e[t1][t2]=t3;
e[t2][t1]=t3;
}
//初始化dis数组,这是一号顶点到各个顶点的厨师距离,应为当前生成树中只有一号顶点
for(i=1;i<=n;i++)
{
dis[i]=e[1][i];
}
//开始prime核心算法
//江一号顶点加入生成树
book[1]=1; //使用book数组来标记一个顶点是否已经加入生成树
count++;
while(count<n)
{
min=inf;
for(i=1;i<=n;i++)
{
if(book[i]==0 && dis[i]<min)
{
min=dis[i];
j=i;
}
}
book[j]=1;count++; sum=sum+dis[j];
//扫描当前顶点j所有的边,再以j为中心点,跟新生成树到每一个非树顶点的距离;
for(k=1;k<=n;k++)
{
if(book[k]==0 && dis[k]>e[j][k])
{
dis[k]=e[j][k];
}
}
}
printf("%d",sum);
return 0;
}
prim算法是以点为中心,边只是用来判断当前最近的非树顶点的权值。
1,用无向图储存需要的信息,n为顶点数,m为边数,e【】【】储存顶点与顶点关系,inf代表未联通,0代表是同一点。
2,dis数组储存当前树集合(所有)到到非树顶点(所有)的权值(距离)。//注意是树集合是包含所有进入生成树的顶点
3,找到目前距离树集合最近的非树顶点,加入树中。利用刚进入树中的顶点将树集合距非树顶点的距离更新一遍。
4,重复n-1遍,将所有的非树顶点加入树集合。
关于第三步注解
1 找到最近的非树顶点,是利用刚进入树集合的顶点与所有非树顶点的距离,即新的树集合距所有非树顶点的距离。
2 if( 新树集合距所有非树顶点的距离 < 原来树集合距所有非树顶点的距离 ) 更新dis数组。
3 利用dis数组找到最近的非树顶点,将其加入树集合。