迪杰斯特拉算法--堆优化版(详细代码注释)

#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>

using namespace std;
const int N = 1e6 + 10, INF = 0x3f3f3f3f; //边数点数均为1e5；
int n, m;   //点数和边数
//采用邻接表建图，因为是稀疏图!
int h[N], e[N], ne[N], w[N], idx;
//标记这个点是否已经确定了最短路径。
bool st[N];     
int dist[N];
//用二元组：小根堆维护的是节点的距离，但是我们还要记录节点编号，所以:<到源点的距离,节点编号>
typedef pair<int, int> PII;

void add (int a, int b, int c)
{
    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++;
}

int dj ()
{
    memset(dist, 0x3f3f3f3f, sizeof dist);
    dist[1] = 0;    //起点到起点的距离为0；
    priority_queue<PII, vector<PII>, greater<PII>> heap;
    heap.push({0, 1});
    
    while (heap.size())
    {
        auto t = heap.top();
        heap.pop();
        
        //将最小距离的节点出堆。
        int v = t.second;
        
     //看是否冗余
        if (st[v]) continue;      
        st[v] = true;
//        遍历邻接表：
        for (int i = h[v]; ~i; i = ne[i])
        {
            int j = e[i];
            if (dist[v] + w[i] < dist[j])
            {
                dist[j] = min(dist[v] + w[i], dist[j]);
                heap.push({dist[j], j});
            }
        }
    }
    
    if (dist[n] == 0x3f3f3f3f) return -1;
    return dist[n];
}

int main()
{
    memset(h, -1, sizeof h);
    cin >> n >> m;
    while (m -- ){
        int a, b, c;
        cin >> a >> b >> c;
        add (a, b, c);
    }
    
    cout << dj() << endl;
    return 0;
}