Dijkstra求最短路 II(堆优化)

题目入口
题意：给定一个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^5,
图中涉及边长均不小于0，且不超过10000。

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

思路：
其实堆优化的Dijkstra和上一篇博客朴素的Dijkstra基本原理是一样的。
只不过朴素的是找到最小边后去更新n个点，但如果n比较大时O(n^2)的效率就不行了。
这个时候就可以找到最小边后去更新m条边，而且为了提高效率还可以用堆即优先队列来寻找，这样找最小值的效率是差不多logn的， 整个堆优化的Dijkstra效率就变成了O(mlogn)。

代码实现：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <cctype>
#include <cstring>
#include <iostream>
#include <sstream>
#include <string>
#include <list>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <algorithm>
#include <functional>
#define lowbit(x) (x &(-x))
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pll;
typedef pair<int, int> pii;
const int  INF = 0x3f3f3f3f;
const double PI = acos(-1.0);
const double EXP = 1e-8;
const ll   MOD = 1e9 + 7;
const int  N = 1e6 + 5;

int n, m;
int h[N], e[N], ne[N], w[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()
{
    mem(dist, 0x3f);
    dist[1] = 0;

    priority_queue<pii, vector<pii>, greater<pii> > heap;
    //先把1号点加入队列
    heap.push({0, 1});
    while(heap.size()){
        //直接取出对头min
        auto t = heap.top(); heap.pop();

        int ver = t.second, dt = t.first;
        //如果已经是确定了最短路的点就跳过
        if(st[ver]) continue;
        st[ver] = 1;
        //开始更新以对头t为起点的边
        for(int i = h[ver]; ~i; i = ne[i]){
            int j = e[i];
            if(dist[j] > dist[ver] + w[i]){
                dist[j] = dist[ver] + w[i];
                heap.push({dist[j], j});
            }
        }
    }

    if (dist[n] == 0x3f3f3f3f) return -1;
    return dist[n];
}

signed main()
{
    IOS;
    cin >> n >> m;
    mem(h, -1);
    while(m --){
        int a, b, c;
        cin >> a >> b >> c;
        add(a, b, c);
    }
    cout << Dijkstra() << endl;
    return 0;
}