最小生成树：Kruskal算法题目

题目链接

题目

农夫约翰被选为他们镇的镇长！

他其中一个竞选承诺就是在镇上建立起互联网，并连接到所有的农场。

约翰已经给他的农场安排了一条高速的网络线路，他想把这条线路共享给其他农场。

约翰的农场的编号是1，其他农场的编号是 2∼n。

为了使花费最少，他希望用于连接所有的农场的光纤总长度尽可能短。

你将得到一份各农场之间连接距离的列表，你必须找出能连接所有农场并使所用光纤最短的方案。

输入

第一行包含一个整数 n，表示农场个数。

接下来 n 行，每行包含 n 个整数，输入一个对角线上全是0的对称矩阵。
其中第 x+1 行 y 列的整数表示连接农场 x 和农场 y 所需要的光纤长度。

输出

输出一个整数，表示所需的最小光纤长度。

数据范围

输入样例

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);
}