将带环图生成为边权和最小的树:最小生成树
题目链接:https://www.luogu.com.cn/problem/P3366
Kruskal算法:并查集,将所有边按权值从小到大排序,从小到大遍历。
如果一条边对应两个顶点不在一个集合内则将其合并,并统计合并次数
若最终合并次数小于(n为顶点数)n - 1,则该图不连通。
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 5010, M = 2e5 + 10;
int n, m;
int p[N];
struct Edge {
int a, b, w;
bool operator<(Edge& t) {
return w < t.w;
}
}edge[M];
int find(int x) {
if (x != p[x])
p[x] = find(p[x]);
return p[x];
}
int kruskal() {
sort(edge, edge + m);
// 初始化并查集
for (int i = 1; i <= n; i++)
p[i] = i;
int res = 0, cnt = 0;
for (int i = 0; i < m; i++) {
int a = edge[i].a, b = edge[i].b, w = edge[i].w;
a = find(a);
b = find(b);
// 合并
if (a != b) {
p[a] = b;
res += w;
cnt++;
}
}
if (cnt < n - 1)return -1;
return res;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
int ta, tb, tw;
cin >> n >> m;
for (int i = 0; i < m; i++) {
cin >> ta >> tb >> tw;
edge[i].a = ta;
edge[i].b = tb;
edge[i].w = tw;
}
int ans = kruskal();
if (ans == -1)
cout << "orz\n";
else cout << ans << '\n';
return 0;
}