最小生成树---prim算法

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;
}