prim算法

/*
Prim算法：
1.从任意一个顶点开始构造生成树，假设就从1号顶点吧，首先将1号顶点加入生成树,
用一个一维数组book标记哪些顶点已经加入了生成树 
2.用数组dis记录生成树到各个顶点的距离。最初生成树中只有1号顶点，有直连边时，
数组dis中存储的是1号顶点到该顶点的边的权值，没有直连边时就是无穷大
3.从数组dis中选出离生成树最近的顶点（假设该顶点为J）加入到生成树中（即在dis中找出最小值） 
再以J为中间点，更新生成树到每一个非树顶点的距离，即如果dis[k]>e[j][k]则更新dis[k]=e[j][k] 
*/ 
#include <stdio.h>

int main()
{
	int n,m,i,j,k,min,t1,t2,t3;
	int e[7][7], dis[7],book[7] = {0};
	int inf = 9999999; //inf(infinity的缩写)存储一个我们认为的正无穷值
	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); //t3为顶点t1,t2之间的权值 
		e[t1][t2] = t3;
		e[t2][t1] = t3;
	 }
	 
	 //初始化dis数组，这里是1号顶点到各个顶点的初始距离，因为当前生成树中只有1号顶点
	 for(i=1;i<=n;i++)
	 {
	 	dis[i] = e[1][i];  //此时生成树中只有1号顶点，数组中存储的就是所有顶点到1号顶点的距离 
	  } 
	  
	  //prim核心部分
	  //将1号顶点加入生成树
	  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);
		  
	getchar();getchar();
	return 0;  
}