hihoCoder-1097(最小生成树)


                                                最小生成树一·Prim算法
 

描述

最近,小Hi很喜欢玩的一款游戏模拟城市开放出了新Mod,在这个Mod中,玩家可以拥有不止一个城市了!

但是,问题也接踵而来——小Hi现在手上拥有N座城市,且已知这N座城市中任意两座城市之间建造道路所需要的费用,小Hi希望知道,最少花费多少就可以使得任意两座城市都可以通过所建造的道路互相到达(假设有A、B、C三座城市,只需要在AB之间和BC之间建造道路,那么AC之间也是可以通过这两条道路连通的)。
提示:不知道为什么Prim算法和Dijstra算法很像呢Σ(っ °Д °;)っ 。

输入

每个测试点(输入文件)有且仅有一组测试数据。

在一组测试数据中:

第1行为1个整数N,表示小Hi拥有的城市数量。

接下来的N行,为一个N*N的矩阵A,描述任意两座城市之间建造道路所需要的费用,其中第i行第j个数为Aij,表示第i座城市和第j座城市之间建造道路所需要的费用。

对于100%的数据,满足N<=10^3,对于任意i,满足Aii=0,对于任意i, j满足Aij=Aji, 0<Aij<10^4.

输出

对于每组测试数据,输出1个整数Ans,表示为了使任意两座城市都可以通过所建造的道路互相到达至少需要的建造费用。

样例输入

    5
    0 1005 6963 392 1182
    1005 0 1599 4213 1451
    6963 1599 0 9780 2789
    392 4213 9780 0 5236
    1182 1451 2789 5236 0

样例输出

    4178

程序代码:

#include<stdio.h>
#include<string.h>
#define inf 0x7f7f7f7f
int e[1010][1010],book[1010],dis[1010];
int main()
{
	int a,n,i,j,count,sumn,minn,u;
	while(scanf("%d",&n)!=EOF)
	{
		count=0;
		sumn=0;
		memset(book,0,sizeof(book));
		for(i=1;i<=n;i++)
			for(j=1;j<=n;j++)
			{
				scanf("%d",&a);
				e[i][j]=a;
			}
		for(i=1;i<=n;i++)
			dis[i]=e[1][i];
		book[1]=1;	
		count++;//代表点 
		while(count<n)
		{
			minn=inf;	
			for(i=1;i<=n;i++)
				if(book[i]==0&&dis[i]<minn)
				{
					minn=dis[i];
					u=i;
				}
			book[u]=1;
			count++;
			sumn+=dis[u];
			for(i=1;i<=n;i++)
				if(book[i]==0&&dis[i]>e[u][i])
					dis[i]=e[u][i];		
		}	
		printf("%d\n",sumn);
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/HeZhiYing_/article/details/81289012