Prime算法:
1:从任意一个顶点开始构造生成树,假设就从0号顶点吧,首先将顶点1号加入生成树中,用一个一维数组book来标记那些顶点已经加入了生成树。
2:用数组dis记录生成树到各个顶点的距离。最初生成树中只有1号顶点,有直连边时,数组dis中存储的就是1号顶点到该顶点的边的权值,没有直连边的时候就是无穷大,即初始化dis数组。
3:从数组dis中选出离生成树最近的顶点(假设这个顶点为j)加入到生成树中(即在数组dis中找最小值)。再以j为中间点,更新生成树到每一个非树顶点的距离(就是松弛),即如果dis[k]>e[j][k]则更新dis[k]=e[j][k]。
4:重复第三步,直到生成树中有n个顶点为止。
代码模板:
#include <stdio.h>
#include <string.h>
#define inf 99999999
int e[110][110];
int book[110],dis[110];
int main()
{
int i,j,k,n,q,a,b,min,count,sum;
while(scanf("%d",&n)!=EOF)
{
count=0;sum=0;
memset(e,0,sizeof(e));
memset(book,0,sizeof(book));
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
scanf("%d",&e[i][j]);
scanf("%d",&q);
for(i=1;i<=q;i++)
{
scanf("%d%d",&a,&b);
e[a][b]=0;
e[b][a]=0;
}
for(i=1;i<=n;i++)
dis[i]=e[1][i];
book[1]=1;
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+=dis[j];
for(k=1;k<=n;k++)
{
if(book[k]==0&&dis[k]>e[j][k])
dis[k]=e[j][k];
}
}
printf("%d\n",sum);
}
return 0;
}