【最短路——堆优化版的Dijkstra算法】AcWing 850. Dijkstra求最短路 II

AcWing 850. Dijkstra求最短路 II

首先：堆优化版的Dijsktra算法适合稀疏图。

堆优化版的Dijkstra算法，对朴素版的Dijkstra算法中步骤2.1、2.3，使用堆（优先队列）的数据结构对其进行了实现。另外，因为是稀疏图，存储图采用邻接表的方式实现。

具体来看的话：

1. 朴素版的时间复杂度主要来自于步骤2.1，寻找距离最近的点t时遍历全部节点n，时间复杂度是O(n)。而使用堆时，这个操作是O(1)的。
2. 朴素版步骤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;
}