杭电OJ 1233 还是畅通工程

杭电OJ 1233 还是畅通工程

题目链接

Problem Description
某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。(当N为0时，输入结束，该用例不被处理。)

Output
对每个测试用例，在1行里输出最小的公路总长度

这道题需要注意测试n=1的情况，当输出distance的语句写在union的for循环里的时候这种情况不能输出预期的0。

#include<bits/stdc++.h>
using namespace std;
vector<int> parent(110);
struct Edge
{
    int a_{ -1 };
    int b_{ -1 };
    int distance_{ -1 };
};
bool cmp(const Edge &lhs, const Edge &rhs)
{
    return lhs.distance_ < rhs.distance_;
}
int find(int x) {
    if (parent[x] != x) parent[x] = find(parent[x]);
    return parent[x];
}
void Union(int x, int y)
{
    int a = find(x);
    int b = find(y);
    if (a != b) parent[a] = b;
}
int main()
{
#ifdef LOCAL
    freopen("input.txt", "r", stdin);//freopen("output.txt", "w", stdout);
#endif
    int n;
    while (cin >> n, n) {
        //边按权值排序.
        int m = n * (n - 1) / 2;
        vector<Edge> edge_set(m);
        for (auto &i : edge_set) cin >> i.a_ >> i.b_ >> i.distance_;
        sort(edge_set.begin(), edge_set.end(), cmp);
        //初始化.
        for (int i = 1; i <= n; ++i) {
            parent[i] = i;
        }
        int distance = 0, ctr = 0;
        //生成最小生成树
        for (auto &i : edge_set) {
            if (find(i.a_) != find(i.b_)) {
                Union(i.a_, i.b_);
                ++ctr;
                distance += i.distance_;
            }
            if (ctr == n - 1) break;
        }
        cout << distance << endl;
    }
}