公路村村通c++prime算法
时间: 2023-11-24 07:50:22 浏览: 101
公路村村通问题可以使用Prime算法来解决。该算法是一种贪心算法,用于寻找加权无向图的最小生成树。
以下是使用C++实现的公路村村通问题的Prime算法:
```c++
#include <iostream>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;
const int MAXN = 1005;
const int INF = 0x3f3f3f3f;
struct Edge {
int to, w;
Edge(int to, int w) : to(to), w(w) {}
};
vector<Edge> G[MAXN];
int d[MAXN];
bool vis[MAXN];
int prim(int s, int n) {
memset(d, INF, sizeof(d));
memset(vis, false, sizeof(vis));
d[s] = 0;
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> q;
q.push(make_pair(0, s));
int ans = 0;
while (!q.empty()) {
int u = q.top().second;
q.pop();
if (vis[u]) continue;
vis[u] = true;
ans += d[u];
for (int i = 0; i < G[u].size(); i++) {
int v = G[u][i].to;
int w = G[u][i].w;
if (!vis[v] && d[v] > w) {
d[v] = w;
q.push(make_pair(d[v], v));
}
}
}
return ans;
}
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));
}
int ans = prim(1, n);
cout << ans << endl;
return 0;
}
```
阅读全文