【最小生成树专题】hihocoder 1097 最小生成树一·Prim算法

http://hihocoder.com/problemset/problem/1097

#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

思路

题目大意：给你一张无向带权完全图，求最小生成树

题目思路：因为题目要求的是用朴素的prim做法来解决这样一道题，所以我们采用朴素的prim做法。

时间复杂度&&空间复杂度：O（n*n）&&O（n*n）（n为点的个数）

AC Code

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
typedef long long ll;
const int nmax=1e3+10;
int G[nmax][nmax];
int d[nmax],vis[nmax];//d[i]每个点i到有效集S的距离 
int n;
int ans,End;//ans边权和,End为每次更新S时新加入的点 

int main(int argc, char** argv) {
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			scanf("%d",&G[i][j]);
		}
	}
	memset(vis,0,sizeof(vis));
	memset(d,0x3f3f3f3f,sizeof(d));
	d[1]=0;
	ans=0;//边权和
	for(int i=1;i<=n;i++){
		End=-1; 
		for(int j=1;j<=n;j++){//找到n条边中距离集合S最近的边 
			if(!vis[j]&&(End==-1||d[End]>d[j])){
				End=j;
			}
		}
		ans+=d[End];
		vis[End]=1;
		for(int j=1;j<=n;j++){//更新集合S到n个点的距离 
			if(!vis[j]&&d[j]>G[End][j]){
				d[j]=G[End][j];
			} 
		}
	} 
	printf("%d\n",ans);
	return 0;
}