考研机试真题--还是畅通工程--浙江大学

关键字：最小生成树：使用kruskal算法，并查集实现
题目：
某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。
输入描述:
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。
输出描述:
对每个测试用例，在1行里输出最小的公路总长度。
示例1
输入
3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0
输出
3
5

链接：
https://www.nowcoder.com/practice/d6bd75dbb36e410995f8673a6a2e2229?tpId=40&tqId=21479&tPage=1&rp=1&ru=/ta/kaoyan&qru=/ta/kaoyan/question-ranking

代码：

#include <iostream>
#include <fstream>
#include <algorithm>
using namespace std;
const int maxn = 110;
const int maxm = 6000;
#define INF 9999999
struct Edge{
    int a, b;
    int weight;
}edge[maxm];
int father[maxn];

bool cmp(Edge x, Edge y){
    return x.weight < y.weight;
}

int findFather(int x){
    if(x == father[x]) return x;
    return findFather(father[x]);
}

int main(){
    int n;
//    freopen("a.txt", "r", stdin);
    while(cin >> n && n != 0){
        int m = n * (n - 1) / 2, sum = 0;
        for(int i = 1; i <= m; ++i){
            cin >> edge[i].a >> edge[i].b >> edge[i].weight;
        }
        sort(edge + 1, edge + m, cmp);
        for(int i = 1; i <= n; ++i){
            father[i] = i;
        }
        for(int i = 1; i <= m; ++i){
            int fa = findFather(edge[i].a);
            int fb = findFather(edge[i].b);
            if(fa != fb){
                father[fa] = fb;
                sum += edge[i].weight;
            }
        }
        cout << sum << endl;
    }
    return 0;
}