AcWing 850. Dijkstra求最短路 II
首先:堆优化版的Dijsktra算法适合稀疏图。
堆优化版的Dijkstra算法,对朴素版的Dijkstra算法中步骤2.1、2.3,使用堆(优先队列)的数据结构对其进行了实现。另外,因为是稀疏图,存储图采用邻接表的方式实现。
具体来看的话:
- 朴素版的时间复杂度主要来自于步骤2.1,寻找距离最近的点t时遍历全部节点n,时间复杂度是O(n)。而使用堆时,这个操作是O(1)的。
- 朴素版步骤2.3更新所有点的距离时,复杂度为O(n),而使用堆来实现时,这个操作为O(mlogn)【堆操作此时是遍历邻近的所有边O(m),然后堆上进行push复杂度O(logn)】。
代码:
#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
typedef pair<int, int> PII;//first为当前节点距起点的距离,second为该节点编号
const int N = 1e6 + 10;
int n, m;
int h[N], w[N], e[N], ne[N], idx;//稀疏图采用邻接表存
int dist[N];//每个点距离起点的距离
bool st[N];//该点是否被处理过
void add(int a, int b, int c) {
e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx++;
}
int dijkstra() {
memset(dist, 0x3f, sizeof dist);//初始化距离
dist[1] = 0;
priority_queue<PII, vector<PII>, greater<PII>> heap;//采用优先队列(堆)寻找当前距离最小的节点,复杂度为O(1)
heap.push({0, 1});
while (heap.size()) {
auto t = heap.top();//从堆里拿出当前距起点最小的节点及其距离
heap.pop();
int ver = t.second, distance = t.first;
if (st[ver]) continue;//当前节点已经处理过的话,跳过。
st[ver] = true;
//遍历t相邻的所有边,更新邻点距离
for (int i = h[ver]; i != -1; i = ne[i]) {
int j = e[i];
if (dist[j] > distance + w[i]) {
//使用 起点到t加t到j的距离 更新 起点到j的距离
dist[j] = distance + w[i];
heap.push({dist[j], j});
}
}
}
if (dist[n] == 0x3f3f3f3f) return -1;
return dist[n];
}
int main() {
scanf("%d%d", &n, &m);
memset(h, -1, sizeof h);//初始化邻接表
while (m--) {
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
add(a, b, c);
}
cout << dijkstra() << endl;
return 0;
}