Dijkstra求最短路Ⅱ|堆优化 稀疏矩阵优化

Dijkstra求最短路Ⅱ|堆优化 稀疏矩阵优化

---

from acwing850

Time limit：1s

Memory limit：64MB

---

Problem Description

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

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

Input

第一行包含整数n和m。

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

Output

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

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

数据范围

1≤n,m≤1.5×10⁵ ,1≤n,m≤1.5×10⁵,
图中涉及边长均不小于0，且不超过10000。

Sample Input

3 3
1 2 2
2 3 1
1 3 4

Sample Output

3

---

对于这个题目，和Dijkstra求最短路Ⅰ相比，数据量变大了，且点边的量级也改变了，对于一般最短路问题，边是点量级的平方，那么一般数据量都很小，可以直接开辟一个二维数组；但是对于大一点的数据量，边和点是相同量级的时候，内存是不够的开辟的，那么我们就要用到稀疏矩阵，或者vector模拟链表（指针）也可以。

（这里要自闭一下，这学期刚学习了稀疏矩阵，结果差点都忘了，还想了好一会儿，不过这个题目也就用到了存储而已，然后加了一点并查集的思想用来索引）

当然，传统朴素dijkstra算法的时候复杂度是O(n^2)的，这里我们利用优先队列降低时间复杂度，并且按照权值从小到大排序（greater），这样每次弹出的元素都是未被标记的点中，离1（起点）最近的，这个操作相当于dijkstra朴素方法中遍历了1到n找最小距离点。

AC代码：

#include<cstdio>
#include<queue>
#include<cstring>
#define pii pair<int,int>
using namespace std;
int n,m;
int from[150005],father[150005],dis[150005],edge = 0;pii p[150005];
bool note[150005];
void init(){
    scanf("%d %d",&n,&m);
    memset(from,-1,sizeof(from));   //后续操作当遇到-1的时候结束dijsktra内部循环
    int f,t,w;
    while(m--){
        scanf("%d %d %d",&f,&t,&w);
        p[edge].first = w,p[edge].second = t,father[edge] = from[f],from[f] = edge++;
    }//这个步骤，是用pair存输入的边的终点（second）和边权（first）
     //father可以理解为并查集，对于所有起点相同的边，下一个位置，存着上一个的索引，而from是用来索引最后一个	   //father的，这里可以和并查集的查找函数联系起来理解
    for(int i = 2;i <= n;++i)
        dis[i] = 0x3f3f3f3f;
}
int main(){
    init();
    priority_queue<pii,vector<pii>,greater<pii> > q;
    q.push({0,1});          //优先队列，按边权从小到大排序，所以first存的是边权
    while(!q.empty()){
        pii x = q.top();q.pop();
        if(note[x.second])
            continue;
        note[x.second] = true;
        for(int i = from[x.second];i != -1;i = father[i])   //经典并查集思想了
            if(!note[p[i].second] && dis[p[i].second] > dis[x.second] + p[i].first)
                dis[p[i].second] = dis[x.second] + p[i].first,q.push({dis[p[i].second],p[i].second});//更新了就压入队列
    }
    if(dis[n] == 0x3f3f3f3f)
        printf("-1");
    else
        printf("%d",dis[n]);
    return 0;
}