acwing 850 Dijkstra求最短路 II

思路与849几乎一样，本文用堆写的方法适合于稀疏图，而849的写法适合稠密图。

#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int N = 1e6 + 10;

typedef pair<int,int> PII;
int n,m;
int h[N],ne[N],e[N],w[N],idx;
int d[N];
bool st[N];


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

int dijkstra()
{
    memset(d, 0x3f, sizeof d);
    d[1] = 0;
    
    priority_queue<PII,vector<PII>,greater<PII>> heap;//小根堆
    heap.push({0,1});//距离&节点编号
    
    while(heap.size())
    {
        auto t = heap.top();//取得离原点最近的点
        heap.pop();
        
        int ver = t.second, distance = t.first;//ver是节点的编号，distance是原点到ver的距离
        if(st[ver]) continue;//如果当前点是距离确定的点，则跳过该点
        st[ver] = true;
        
        for(int i = h[ver]; i != -1;i = ne[i])
        {
            int j = e[i];
            if(d[j] > d[ver] + w[i])
            {
                d[j] = d[ver] + w[i];
                heap.push({d[j],j});//当距离变小时就入堆
            }
        }
    }
    if(d[n] == 0x3f3f3f3f)
    return -1;
    return d[n];
}

int main()
{
    cin >> 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;
}