HDU 1233 - 还是畅通工程 （Prim）

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。

Output

对每个测试用例，在1行里输出最小的公路总长度。

Sample Input

3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0

Sample Output

3
5

题意：

      求村庄之间连同所修路的最小值。

思路：

      最小生成树，Prim。

代码：

#include<stdio.h>
#include<string.h>
#define inf 99999999
int map[5000][5000],dis[5000],book[5000];
int main()
{
	int n,m,i,j,k,min,a,b,c,count,sum;
	while(scanf("%d",&n)!=EOF&&n)
	{
		count=sum=0;
		memset(book,0,sizeof(book));
		memset(dis,0,sizeof(dis));
		m=n*(n-1)/2;
		for(i=1;i<=n;i++)
			for(j=1;j<=n;j++)
			{
				if(i==j)
					map[i][j]=0;
				else
					map[i][j]=inf;
			}
		for(i=1;i<=m;i++)
		{
			scanf("%d%d%d",&a,&b,&c);
			if(c<map[a][b])
				map[a][b]=map[b][a]=c;
		}
		for(i=1;i<=n;i++)
			dis[i]=map[1][i];
		book[1]=1;
		dis[1]=0;
		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]>map[j][k])
					dis[k]=map[j][k];
			}
		}
		printf("%d\n",sum);
	}
}