-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path1029.cpp
More file actions
114 lines (99 loc) · 1.74 KB
/
Copy path1029.cpp
File metadata and controls
114 lines (99 loc) · 1.74 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
//最小生成树
#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
using namespace std;
int m, n, E; //m组测试数据,顶点数、边数
struct edge
{
int u; // 起点
int v; // 终点
int w; // 无向边的权值
// 构造函数
edge(int start, int end, int weight)
{
u = start;
v = end;
w = weight;
}
};
// 并查集数据结构
struct UnionFind {
vector<int> parent;
vector<int> rank;
UnionFind(int n) {
parent.resize(n + 1);
rank.resize(n + 1, 0);
for (int i = 1; i <= n; ++i) {
parent[i] = i;
}
}
int find(int x) {
if (parent[x] != x) {
parent[x] = find(parent[x]);
}
return parent[x];
}
void unite(int x, int y) {
int rootX = find(x);
int rootY = find(y);
if (rootX == rootY) {
return;
}
if (rank[rootX] < rank[rootY]) {
parent[rootX] = rootY;
}
else if (rank[rootX] > rank[rootY]) {
parent[rootY] = rootX;
}
else {
parent[rootY] = rootX;
rank[rootX]++;
}
}
};
bool cmp(const edge& a, const edge& b) {
return a.w < b.w;
}
int kruskal(vector<edge>& edges) {
UnionFind uf(n);
sort(edges.begin(), edges.end(), cmp);//仿函数改变sort规则
int totalWeight = 0;
int numEdges = 0;
for (int i = 0; i < edges.size(); i++) {
const edge& e = edges[i];
if (uf.find(e.u) != uf.find(e.v)) {
uf.unite(e.u, e.v);
totalWeight += e.w;
numEdges++;
}
if (numEdges == n - 1) {
break;
}
}
if (numEdges == n - 1) {
return totalWeight;
}
else {
return -1;
}
}
int main() {
cin >> m;
while (m--)
{
cin >> n;//顶点数、边数、顶点s以及顶点t.
cin >> E;
vector<edge>edges;
int u1, v1, w1;
for (int i = 0; i < E; i++) {//输入无向边的信息
cin >> u1;
cin >> v1;
cin >> w1;
edges.push_back(edge(u1, v1, w1));
}
cout << kruskal(edges) << endl;
}
return 0;
}