题目描述:
农夫约翰被选为他们镇的镇长!
他其中一个竞选承诺就是在镇上建立起互联网,并连接到所有的农场。
约翰已经给他的农场安排了一条高速的网络线路,他想把这条线路共享给其他农场。
约翰的农场的编号是1,其他农场的编号是 2∼n。
为了使花费最少,他希望用于连接所有的农场的光纤总长度尽可能短。
你将得到一份各农场之间连接距离的列表,你必须找出能连接所有农场并使所用光纤最短的方案。
输入格式
第一行包含一个整数 n,表示农场个数。
接下来 n 行,每行包含 n 个整数,输入一个对角线上全是0的对称矩阵。
其中第 x+1 行 y 列的整数表示连接农场 x 和农场 y 所需要的光纤长度。
输出格式
输出一个整数,表示所需的最小光纤长度。
数据范围
3≤n≤100
每两个农场间的距离均是非负整数且不超过100000。
输入样例:
4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0
输出样例:
28
分析:
本题顶点数较小,可以用prim算法去求最小生成树,时间复杂度是O(n^2)。
prim算法与dijkstra算法很相似,不过dijkstra算法的距离数组d[i]表示的是i到起点的最短距离,而prim算法的d[i]表示的则是i到已经构造好的生成树的最近距离。prim算法的过程也很简单,n个点,利用n次循环,每次向生成树里加入一个点。初始时选定一个点作为生成树最原始的点,然后用这个点去更新周围点的距离数组,后面的循环中要找出还未加入生成树的点集中离已经在生成树中点集最近的点,连上对应的边,将该点加入到生成树中,同时用这点去更新周围点的d数组即可。
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 105;
int n,g[N][N],d[N];
bool st[N];
int prim(){
int res = 0;
memset(d,0x3f,sizeof d);
d[0] = 0;
for(int i = 0;i < n;i++){
int t = -1;
for(int j = 0;j < n;j++){
if(!st[j] && (t == -1 || d[t] > d[j]))
t = j;
}
st[t] = true;
res += d[t];
for(int j = 0;j < n;j++) d[j] = min(d[j],g[t][j]);//与dijkstra算法的不同之处
}
return res;
}
int main(){
cin>>n;
for(int i = 0;i < n;i++)
for(int j = 0;j < n;j++)
cin>>g[i][j];
cout<<prim()<<endl;
return 0;
}