dijkstra求最短路2

简介: 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×105,
图中涉及边长均不小于 0,且不超过 10000。
数据保证:如果最短路存在,则最短路的长度不超过 109。

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

#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;

const int N=1e6+10;
typedef pair<int, int> pii;

int h[N], e[N], ne[N], w[N], idx;
int d[N];
int n, m;
bool st[N];

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

void 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, dist=t.first;
        if(st[ver]) continue;
        st[ver]=true;
        for(int i=h[ver];i!=-1;i=ne[i])
        {
            int j=e[i];
            if(d[j]>dist+w[i])
            {
                d[j]=dist+w[i];
                heap.push({d[j], j});
            }
        }
    }
}

int main()
{
    scanf("%d %d", &n, &m);
    memset(h, -1, sizeof h);
    while(m--)
    {
        int a, b, c;
        scanf("%d %d %d", &a, &b, &c);
        add(a, b, c);
    }
    dijkstra();
    if(d[n]==0x3f3f3f3f) printf("-1");
    else printf("%d", d[n]);
    return 0;
}
相关文章
|
6月前
|
算法
最短路之Floyd算法
最短路之Floyd算法
74 1
|
6月前
|
算法
最短路之Dijkstra算法
最短路之Dijkstra算法
54 0
|
6月前
|
存储 算法
Dijkstra
Dijkstra“【5月更文挑战第18天】”
54 6
|
算法
SPFA算法-最短路-负环
SPFA算法-最短路-负环
84 0
|
6月前
|
存储 算法
最短路之SPFA算法
最短路之SPFA算法
54 0
|
11月前
|
C++
|
11月前
|
机器学习/深度学习 编解码 算法
|
算法
dijkstra最短路算法
dijkstra最短路算法
|
机器学习/深度学习 定位技术