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。
数据保证：如果最短路存在，则最短路的长度不超过(10)^9。

输入样例：

3 3
1 2 2
2 3 1
1 3 4

输出样例：

3

题解

#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
//定义一个PII数据类型，其值有两个int类型
//第一个int表示从出发节点1到该节点的距离
//第二个int表示该节点为几
typedef pair<int,int> PII;
int n,m;//点数n，边数m
const int N = 1e6 + 10;
//邻接表
/*
 * h[N]：为节点N到其他节点数据的链表头结点
 * w[N]：存放权值，即c
 * e[N]：存放目标节点，即b
 * a：表示出发节点，也作为头结点数组h[]的索引
 * ne[N]：存放同节点到另一节点的边的地址索引
 * idx：链接w[]，e[]和ne[]，并且指示所用数组
*/
int h[N],w[N],e[N],ne[N],idx;
int dist[N];//解集
bool st[N];//该节点是否扩展过

//将输入的数据a,b,c添加进邻接表
void add(int a,int b,int c){
    ne[idx]=h[a];
    h[a]=idx;
    e[idx]=b;
    w[idx]=c;
    idx++;
}

int Dijkstra(){
    //将解集dist[]的所有数值初始化为无穷大
    memset(dist,0x3f,sizeof(dist));
    //节点1到节点1的距离为0
    dist[1]=0;
    //greater：将优先队列heap设置为小顶堆，即从小到大排序，数值小的在前面
    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;
        //对于扩展过的点，将采取跳过，不会重复扩展
        if (st[ver]){continue;}
        //标记该点已被扩展
        st[ver] = true;
        for (int i = h[ver]; i != -1 ; 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];
}

int main(){
    cin >> n >> m;
    int a,b,c;
    //将头结点数组都初始化为-1
    memset(h,-1,sizeof(h));
    while(m--){
        cin >> a >> b >> c;
        add(a,b,c);
    }
    cout << Dijkstra() << endl;
}

补充：

1、if(st[ver]){continue;}的作用是，当该点已被确认为最小距离时，不会在进行重复运算。if (dist[j] > dist[ver]+w[i])作用为只有距离小时才更新，因此算法流程，所以在队列中会有不同距离的同个点，heap为小顶堆导致此点只有最小距离会被弹出运算，即最小距离。两步结合的总体作用为只保证一个点的最小距离会参与运算，避免多余计算。

2、关于能用if(!st[j] && (dist[j] > dist[ver]+w[i]))代替上述两行的错误理解：该步与if (dist[j] > dist[ver]+w[i])并无不同，因为算法流程，能加入该点时，必然是该点并未被标记，所以加入的还是不同距离的同个点，并不能避免后续该点的多余计算。该步骤中的!st[j]的真实意义为当该点已被确认为最小距离点时，会拒绝后续该点的加入队列。