学习笔记——prim（最小生成树）

代码手抄自《啊哈！算法》

#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数组找到最近的非树顶点，将其加入树集合。