c++分别用Prim算法和Kruskal算法编程实现一下连通无向图的最小生成树。
时间: 2024-03-01 14:55:10 浏览: 95
好的,下面分别给出使用 Prim 算法和 Kruskal 算法实现连通无向图的最小生成树的 C++ 代码实现。
首先,我们需要定义一个边的结构体,用于存储边的起点、终点和权重:
```c++
struct Edge {
int src, dest, weight;
};
```
接下来,我们分别实现 Prim 算法和 Kruskal 算法。
使用 Prim 算法实现最小生成树:
```c++
#include <iostream>
#include <vector>
#include <queue>
#include <climits>
using namespace std;
// 定义边
struct Edge {
int src, dest, weight;
};
// 定义比较函数,用于优先队列
struct Compare {
bool operator()(Edge const& e1, Edge const& e2) {
return e1.weight > e2.weight;
}
};
// 定义 Prim 算法函数
void primMST(vector<vector<int>>& graph, int n) {
vector<int> key(n, INT_MAX);
vector<bool> mstSet(n, false);
priority_queue<Edge, vector<Edge>, Compare> pq;
key[0] = 0;
pq.push({0, 0, 0});
while (!pq.empty()) {
int u = pq.top().dest;
pq.pop();
if (mstSet[u]) continue;
mstSet[u] = true;
for (int v = 0; v < n; v++) {
if (graph[u][v] && !mstSet[v] && graph[u][v] < key[v]) {
key[v] = graph[u][v];
pq.push({u, v, key[v]});
}
}
}
int sum = 0;
for (int i = 1; i < n; i++) {
sum += key[i];
}
cout << "Prim算法得到的最小生成树权值和为 " << sum << endl;
}
// 测试函数
int main() {
int n = 5;
vector<vector<int>> graph = {
{0, 2, 0, 6, 0},
{2, 0, 3, 8, 5},
{0, 3, 0, 0, 7},
{6, 8, 0, 0, 9},
{0, 5, 7, 9, 0}
};
primMST(graph, n);
return 0;
}
```
使用 Kruskal 算法实现最小生成树:
```c++
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
// 定义边
struct Edge {
int src, dest, weight;
};
// 定义比较函数,用于排序
bool compare(Edge const& e1, Edge const& e2) {
return e1.weight < e2.weight;
}
// 定义 Kruskal 算法函数
void kruskalMST(vector<Edge>& edges, int n) {
vector<Edge> result(n - 1);
sort(edges.begin(), edges.end(), compare);
vector<int> parent(n);
for (int i = 0; i < n; i++) {
parent[i] = i;
}
int e = 0, i = 0;
while (e < n - 1) {
Edge next_edge = edges[i++];
int x = parent[next_edge.src];
int y = parent[next_edge.dest];
if (x != y) {
result[e++] = next_edge;
for (int j = 0; j < n; j++) {
if (parent[j] == y) {
parent[j] = x;
}
}
}
}
int sum = 0;
for (int i = 0; i < n - 1; i++) {
sum += result[i].weight;
}
cout << "Kruskal算法得到的最小生成树权值和为 " << sum << endl;
}
// 测试函数
int main() {
int n = 5;
vector<Edge> edges = {
{0, 1, 2},
{0, 3, 6},
{1, 2, 3},
{1, 3, 8},
{1, 4, 5},
{2, 4, 7},
{3, 4, 9}
};
kruskalMST(edges, n);
return 0;
}
```
以上代码中,我们分别实现了 Prim 算法和 Kruskal 算法。在 Prim 算法中,我们使用了一个优先队列来维护当前最小的边,同时使用了一个 `key` 数组来记录每个节点到当前生成树的最小边权值。在 Kruskal 算法中,我们使用了一个并查集来维护当前生成树。
在测试函数中,我们定义了一个无向图,然后分别调用了 Prim 算法和 Kruskal 算法来计算最小生成树的权值和,最终输出结果。
希望这个代码能够帮助到您!
阅读全文