用C++用普里姆算法求最小生成树。
时间: 2024-03-01 15:53:30 浏览: 65
用普里姆算法求最小生成树 MFC
5星 · 资源好评率100%
好的,下面是用C++实现普里姆算法求最小生成树的代码:
```c++
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
#define MAXN 1000 // 最大点数
#define INF 0x3f3f3f3f // 无穷大
struct Edge {
int to, weight;
Edge(int t, int w) : to(t), weight(w) {}
};
vector<Edge> G[MAXN]; // 存储图
int dis[MAXN]; // 存储点到集合S的距离
bool vis[MAXN]; // 存储点是否在集合S中
void prim(int start) {
// 初始化
for (int i = 0; i < MAXN; i++) {
dis[i] = INF;
vis[i] = false;
}
dis[start] = 0;
// 使用优先队列存储点到集合S的距离
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> q;
q.push(make_pair(dis[start], start));
// 普里姆算法主体部分
while (!q.empty()) {
int u = q.top().second;
q.pop();
if (vis[u]) continue; // 如果点u已经在集合S中,则跳过
vis[u] = true; // 将u加入集合S
// 更新与u相邻的点的距离
for (int i = 0; i < G[u].size(); i++) {
Edge e = G[u][i];
int v = e.to, w = e.weight;
if (!vis[v] && dis[v] > w) {
dis[v] = w;
q.push(make_pair(dis[v], v));
}
}
}
}
int main() {
// 读入图的边
int n, m;
cin >> n >> m;
for (int i = 0; i < m; i++) {
int u, v, w;
cin >> u >> v >> w;
G[u].push_back(Edge(v, w));
G[v].push_back(Edge(u, w));
}
// 求最小生成树
prim(1);
// 输出最小生成树的权值和
int ans = 0;
for (int i = 1; i <= n; i++) {
if (dis[i] != INF) ans += dis[i];
}
cout << ans << endl;
return 0;
}
```
其中,我们将图存储在邻接表 `G` 中,使用优先队列存储点到集合S的距离,使用数组 `dis` 存储点到集合S的距离,使用数组 `vis` 存储点是否在集合S中。算法的主体部分是一个循环,每次取出优先队列中距离最小的点,将其加入集合S,并更新与该点相邻的点的距离。这个循环将持续到所有点都加入集合S。最后,我们将数组 `dis` 中小于 `INF` 的值相加,得到最小生成树的权值和。
需要注意的是,普里姆算法的时间复杂度为 $O(m\log n)$,其中 $n$ 是点数,$m$ 是边数。
阅读全文