算法题堆优化版本Dijkstra（Python）

最新推荐文章于 2025-03-25 03:03:05 发布

武倔

最新推荐文章于 2025-03-25 03:03:05 发布

阅读量851

点赞数 1

分类专栏： Python 算法文章标签： python 算法 dijkstra 图论

本文链接：https://blog.csdn.net/blowfire123/article/details/110292935

版权

该博客介绍了如何使用堆优化的Dijkstra算法解决有向图中1号点到n号点的最短路径问题。在给定的1.5×10^5个节点和边的范围内，算法需处理可能存在的重边和自环，且边权非负。当无法从1号点到达n号点时，输出-1。文章提供了一种输入输出格式，并给出具体的数据范围和样例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

给定一个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。

输入样例：

输出样例：

代码：

from heapq import *

n, m = map(int, input().split())
idx = 0
h = [-1]*(n+1)
e, ne, w = [0]*(2*m+1), [0]*(2*m+1), [0]*(2*m+1)
st, dist = [False]*(n+1), [float('inf')]*(n+1)

def add(a, b, wei):
    global idx
    e[idx], w[idx] = b, wei
    h[a], ne[idx] = idx, h[a]
    idx += 1

def dijkstra():
    heap = []
    heappush(heap, [0, 1])
    dist[1]