acwing 850. Dijkstra求最短路（二）堆优化版本

给定一个 n 个点 m 条边的有向图，图中可能存在重边和自环，所有边权均为非负值。

请你求出 1 号点到 n 号点的最短距离，如果无法从 1 号点走到 n 号点，则输出 −1。

输入格式
第一行包含整数 n 和 m。

接下来 m 行每行包含三个整数 x,y,z，表示存在一条从点 x 到点 y 的有向边，边长为 z。

输出格式
输出一个整数，表示 1 号点到 n 号点的最短距离。

如果路径不存在，则输出 −1。

数据范围
1≤n,m≤1.5×10⁵,
图中涉及边长均不小于 0，且不超过 10000。
数据保证：如果最短路存在，则最短路的长度不超过 10⁹。

输入样例：
3 3
1 2 2
2 3 1
1 3 4
输出样例：
3

直接之前朴素版本的修改

1. 这里是稠密图，需要改成邻接表形式存储
2. 用堆进行优化，降低时间复杂度

如何判断稠密图还是稀疏图：
稠密图： 边 等于 点数量的平方
稀疏图 ： 边和点的数量差不多1: 1

code:

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

using namespace std;

typedef pair<int, int> PII;

const int N = 200010;

int h[N], e[N], w[N], ne[N], idx;

int dist[N];  //用于记录每一个点距离第一个点的距离
bool st[N];   //用于记录该点的最短距离是否已经确定

int n,m;  // 用n 表示点数量，用m表示边数量

int add(int a, int b, int c)
{
    w[idx] = c;    // 用来存权重
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx ++;
}

int Dijkstra()
{
    memset(dist, 0x3f, sizeof dist); //初始化距离  0x3f代表无限大
    
    dist[1] = 0; //第一个点到自身的距离为0
    
    priority_queue<PII, vector<PII>, greater<PII>> heap;
    
    // 这个顺序不能倒，pair排序时是先根据first，再根据second，这里显然要根据距离排序
    // {}中存储的是：点的距离最小值，点的边下标（因为用的是邻接表存储，这里存储的是下标）
    heap.push({0, 1});
    
    while(heap.size())
    {
        PII k = heap.top();   // 取不在集合S中距离最短的点
        heap.pop();
        
        int ver = k.second, distance = k.first;
        
        if(st[ver]) continue;
        
        st[ver] = true;
        
        for(int i = h[ver]; i != -1; i = ne[i])
        {
            int j = e[i];// i只是个下标，e中在存的是i这个下标对应的点
            if(dist[j] > distance + w[i])
            {
                dist[j] = distance + w[i];
                heap.push({dist[j], j});
            }
        }
    }
  
    //如果第n个点路径为无穷大即不存在最低路径
    if(dist[n] == 0x3f3f3f3f) return -1;   
    
    return dist[n];
}

int main()
{
    cin >> n >> m;
    
    memset(h, -1, sizeof h);  //初始化图 因为是求最短路径, 所以每个点初始为无限大
    
    while (m -- )
    {
        int a, b, c;
        cin >> a >> b >> c;
        add(a, b, c);
    }
    
    cout << Dijkstra() << endl;
    
    return 0;
}