C++贪心法求最小生成树
时间: 2023-12-23 19:03:55 浏览: 82
贪心算法,最小生成树
C++贪心法求最小生成树的基本思路是Kruskal算法,具体步骤如下:
1.将所有边按照权值从小到大排序。
2.从权值最小的边开始,依次判断这条边的两个端点是否在同一个连通块中,如果不在同一个连通块中,则将这条边加入最小生成树中,并将这两个端点所在的连通块合并为一个连通块。
3.重复步骤2,直到最小生成树中的边数等于总顶点数目减去1或者是测试完所有边时结束。
下面是一个C++实现Kruskal算法求最小生成树的例子:
```c++
#include <iostream>
#include <algorithm>
using namespace std;
const int MAXN = 1005;
const int MAXM = 100005;
struct Edge {
int u, v, w;
} e[MAXM];
int fa[MAXN];
int find(int x) {
if (fa[x] == x) return x;
return fa[x] = find(fa[x]);
}
bool cmp(Edge a, Edge b) {
return a.w < b.w;
}
int main() {
int n, m;
cin >> n >> m;
for (int i = 1; i <= m; i++) {
cin >> e[i].u >> e[i].v >> e[i].w;
}
sort(e + 1, e + m + 1, cmp);
for (int i = 1; i <= n; i++) {
fa[i] = i;
}
int cnt = 0, ans = 0;
for (int i = 1; i <= m; i++) {
int u = e[i].u, v = e[i].v, w = e[i].w;
int fu = find(u), fv = find(v);
if (fu != fv) {
fa[fu] = fv;
cnt++;
ans += w;
if (cnt == n - 1) break;
}
}
if (cnt < n - 1) cout << "No solution" << endl;
else cout << ans << endl;
return 0;
}
```
阅读全文