hihoCoder #1097 最小生成树之Prim算法

原题网址，http://hihocoder.com/problemset/problem/1097

#1097 : 最小生成树一·Prim算法

时间限制:10000ms

单点时限:1000ms

内存限制:256MB

描述

最近，小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

 1 #include <cstdio>
 2 #include <string.h>
 3 #include <algorithm>
 4 using namespace std;
 5 const int N = 1e3+100, INF = 0x3f3f3f3f;
 6 int edge[N][N]; //邻接矩阵
 7 int Prim(int n) //Prim算法
 8 {
 9     bool vis[N] = {0};  //记录已加入的顶点
10     int D[N], MIN, tag, sum = 0;
11     memcpy(D, edge[0], sizeof(D));  //已顶点0为起点初始化数组D
12     vis[0] = true;  //记录顶点0已加入
13     for(int i=1; i<n; ++i)
14     {
15         MIN = 0x3f3f3f3f;   //寻找离已加入的顶点距离最近的点
16         for(int j=1; j<n; ++j)
17             if(!vis[j] && D[j] < MIN)
18                 MIN = D[j], tag = j;
19         vis[tag] = true;    //将找到的最近点加入
20         for(int j=1; j<n; ++j)
21             if(!vis[j])
22                 D[j] = min(D[j], edge[tag][j]); //更新D数组
23     }
24     for(int i=0; i<n; ++i)
25         sum += D[i];
26     return sum;
27 }
28 int main(void)
29 {
30     int n;
31     while(~scanf("%d", &n))
32     {
33         for(int i=0; i<n; ++i)
34             for(int j=0; j<n; ++j)
35                 scanf("%d", &edge[i][j]);   //用邻接矩阵保存图
36         printf("%d\n", Prim(n));
37     }
38 }