编程练习7 5G网络建设

需要用到并查集的相关知识：可以参考如下链接

并查集详解（原理+代码实现+应用+优化）-CSDN博客

#include<iostream>
#include<algorithm>
#include<vector>using namespace std;vector<int> split(string params_str) {vector<int> p;while (params_str.find(" ") != string::npos) {int found = params_str.find(" ");p.push_back(stoi(params_str.substr(0, found)));params_str = params_str.substr(found + 1);}    p.push_back(stoi(params_str));return p;
}vector<string> split_str(string params_str) {vector<string> p;while (params_str.find(" ") != string::npos) {int found = params_str.find(" ");p.push_back(params_str.substr(0, found));params_str = params_str.substr(found + 1);}    p.push_back(params_str);return p;
}  // 并查集实现
class UnionFind{vector<int> root;vector<int> rank;int cnt;public:
// 初始化数据结构UnionFind(int N) : cnt(0){root.resize(N+1);rank.reserve(N+1);for (int i = 0; i < N+1; ++i) {root[i] = i;rank[i] = 1;}}int find(int x) {if (x == root[x]) {return x;}return root[x] = find(root[x]);}void union_op(int x, int y) {root[find(x)] = find(y);cnt+=1;}int get_count(){return cnt;}
};int main()
{int n,m;cin >> n >> m;UnionFind uf(n);vector<vector<int>> networks;for (int i = 0; i < m; i++) {int a,b,c,d;cin >> a >> b >> c >> d;if (d == 1) {if (uf.find(a) != uf.find(b)) {uf.union_op(a, b);}} else {networks.push_back(vector<int>{a,b,c});}}sort(networks.begin(), networks.end(),[](vector<int> a ,vector<int> b){return a[2]<b[2];});int result = 0;int i=0;while(true){if(i>=networks.size()){break;} else {if (uf.find(networks[i][0]) != uf.find(networks[i][1])) {uf.union_op(networks[i][0], networks[i][1]);result += networks[i][2];if (uf.get_count() == n - 1) {break;}}}i+=1;}if(uf.get_count() != n - 1){result = -1; }cout<<result<<endl;return 0;
}