杭电OJ 1233 还是畅通工程

本文主要是介绍杭电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 LOCALfreopen("input.txt", "r", stdin);//freopen("output.txt", "w", stdout);
#endifint 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;}
}