c++ dijkstra算法 时间复杂度0(n logn)
时间: 2023-06-25 10:08:32 浏览: 148
Dijkstra算法是一种用于解决单源最短路径问题的贪心算法。它的时间复杂度为O(n logn),其中n是图中节点的数量。下面是C++实现:
```c++
#include <iostream>
#include <vector>
#include <queue>
#include <climits>
using namespace std;
const int MAXN = 1000;
const int INF = INT_MAX;
vector<pair<int, int>> adj[MAXN];
int dist[MAXN];
void dijkstra(int s, int n) {
for (int i = 0; i < n; i++) {
dist[i] = INF;
}
dist[s] = 0;
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq;
pq.push(make_pair(0, s));
while (!pq.empty()) {
int u = pq.top().second;
int d = pq.top().first;
pq.pop();
if (d > dist[u]) {
continue;
}
for (auto& edge : adj[u]) {
int v = edge.first;
int w = edge.second;
if (dist[u] + w < dist[v]) {
dist[v] = dist[u] + w;
pq.push(make_pair(dist[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;
adj[u].push_back(make_pair(v, w));
adj[v].push_back(make_pair(u, w));
}
dijkstra(0, n);
for (int i = 0; i < n; i++) {
cout << dist[i] << " ";
}
cout << endl;
return 0;
}
```
在上面的实现中,使用了一个优先队列来维护当前到源点距离最小的节点,从而保证每次找到的节点是当前最短路径的节点。由于每个节点最多被加入一次优先队列中,因此总时间复杂度为O(n logn)。
阅读全文