Prim最小生成树-------电子农业网(Agri-Net)

VJ电子农业网(Agri-Net)
题目
农夫约翰当选为本镇市长！他的竞选承诺之一是将互联网连接到该地区的所有农场。他当然需要你的帮助。
农民约翰为他的农场订购了一个高速连接，并将与其他农民分享他的连接。为了将成本降到最低，他想把最小数量的光纤连接到他的农场和所有其他农场。
给出连接每一对农场需要多少纤维的列表，你必须找到将它们连接在一起所需的最小数量的纤维。每个场必须连接到其他场，以便包可以从任何一个场流到任何其他场。
任何两个农场之间的距离不得超过10万。
输入
输入包括几个案例。对于每一种情况，第一行包含农场数N(3<=N<=100)。下面的行包含NxN连续矩阵，其中每个元素显示从农场到另一个单元的距离。从逻辑上讲，它们是N个空间分隔整数的N行.在物理上，他们的长度限制在80个字符，所以有些行继续在其他。当然，对角线将是0，因为从农场I到它本身的距离对这个问题不感兴趣。
输出量
对于每一种情况，输出一个整数长度，即连接整个农场所需的最小光纤长度之和

解答：典型的例题直接prim一次即可求出答案。

#include<iostream>
#include<cstring>
#define inf 0x3f3f3f3f
using namespace std;
int map[105][105];
int used[105];
int dist[105];
int n;
void init()
{
	memset(used,0,sizeof(used));
	memset(dist,inf,sizeof(dist));
}
int  prim()
{
	init();
	int i;
	int sum=0;
	dist[1]=0;
	while(1)
	{
		int u=-1,v,i,min=inf;
		for(i=1;i<=n;i++)
		{
			if(min>dist[i]&&used[i]==0)
			{
				min=dist[i];
				u=i;
			}
		}
		if(u==-1)
		{
			break;
		}
		used[u]=1;
		sum+=dist[u];
		for(v=1;v<=n;v++)
		{
			if(map[u][v]!=inf&&dist[v]>map[u][v]){
				dist[v]=map[u][v];
			}
		}
	}
	return sum;
} 
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		int i,j;
		int sum=0;
		memset(map,0x3f,sizeof(map));
		for(i=1;i<=n;i++)
		{
			for(j=1;j<=n;j++)
			{
				scanf("%d",&map[i][j]);
			}
		}
		sum = prim();
	    cout<<sum<<endl;
	}
	return 0;
}