题目
农夫约翰被选为他们镇的镇长!
他其中一个竞选承诺就是在镇上建立起互联网,并连接到所有的农场。
约翰已经给他的农场安排了一条高速的网络线路,他想把这条线路共享给其他农场。
约翰的农场的编号是1,其他农场的编号是 2∼n。
为了使花费最少,他希望用于连接所有的农场的光纤总长度尽可能短。
你将得到一份各农场之间连接距离的列表,你必须找出能连接所有农场并使所用光纤最短的方案。
输入
第一行包含一个整数 n,表示农场个数。
接下来 n 行,每行包含 n 个整数,输入一个对角线上全是0的对称矩阵。
其中第 x+1 行 y 列的整数表示连接农场 x 和农场 y 所需要的光纤长度。
输出
输出一个整数,表示所需的最小光纤长度。
数据范围
扫描二维码关注公众号,回复:
14230681 查看本文章
![](/qrcode.jpg)
输入样例
4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0
输出样例
28
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1e2 + 10;
struct Edge{
int a, b, w;
bool operator < (const Edge &t) const {
return w < t.w;
}
}e[N * N];
int f[N];
int find(int x){
return f[x] == x ? x : f[x] = find(f[x]);
}
void merge(int a, int b){
f[find(b)] = find(a);
}
int Kruskal(int m){
sort(e, e + m);
int cnt = 0, ans = 0;
for(int i = 0; i < m; i++){
int a = e[i].a, b = e[i].b, w = e[i].w;
int fa = find(a), fb = find(b);
if(fa != fb){
merge(fa, fb);
ans += w;
cnt++;
}
}
return ans;
}
int main(){
int n, m = 0, t;
cin >> n;
for(int i = 1; i <= n; i++) f[i] = i;
for(int i = 1; i <= n; i++){
for(int j = 1; j <= n; j++){
scanf("%d", &t);
if(j > i){
e[m].a = i, e[m].b = j, e[m++].w = t;
}
}
}
cout << Kruskal(m);
}