最短路算法回顾1：Dijkstra（python）

数学人学c语言

已于 2024-04-11 15:33:13 修改

阅读量121

点赞数 1

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

于 2024-04-11 11:28:44 首次发布

本文链接：https://blog.csdn.net/W_extend/article/details/137632134

版权

图专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目是acwing 4871. 最早时刻

首先看数据范围，知时间复杂度最高也不过（nlogn）

看见题目第一行，突然惊觉，如果考试中没说（不含重边和自环），那么数据需要预处理，比如说，如果有两条边从a到b，根据题目要花费最少时间，肯定是保留a到b那条时间少的边。

题目要求计算从1号到n号的最短距离，脑海中就两个（Dijkstra和bellman），当有自环和负权重的图时，使用bellman，其他就Dj吧。

那么主体算法就是Dj为主，题目中说有禁止离开时刻表，又是升序，那么妥妥二分法。

from collections import defaultdict
import heapq as pq

def addtime(site, time):
    ls = no[site]
    if k[site] == 0 or ls[0] > time or ls[-1] < time:
        return time
    l, r = 0, len(ls)-1
    while l < r:
        mid = l + r >> 1
        if ls[mid] < time:
            l = mid + 1
        else:
            r = mid
    if ls[l] > time:
        return time
    while l < len(ls) and ls[l] == time:
        time += 1
        l += 1
    return time


n, m = map(int,input().split())
edge = defaultdict(dict)
k = [0]*(n+1)
no = [ [] ]
for _ in range(m):
    a, b, c = map(int,input().split())
    edge[a][b] = c
    edge[b][a] = c
for idx in range(1, n+1):
    ls = list(map(int,input().split()))
    k[idx] = ls[0]
    no.append(ls[1:])
qu = []
pq.heapify(qu)
pq.heappush(qu, (1, 0))
v = [float('inf')]*(n+1)
while qu:
    site, time = pq.heappop(qu)
    if time > v[site]: continue
    v[site] = time
    time_ = addtime(site, time)
    for nex in edge[site]:
        if edge[site][nex] + time_ < v[nex]:
            pq.heappush(qu, (nex, edge[site][nex] + time_))

if v[-1] == float('inf'):
    print(-1)
else:
    print(v[-1])