杭电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;
    }
}
已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页