結果

問題 No.160 最短経路のうち辞書順最小
ユーザー lloyz
提出日時 2023-06-27 21:40:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 113 ms / 5,000 ms
コード長 782 bytes
コンパイル時間 195 ms
コンパイル使用メモリ 82,328 KB
実行使用メモリ 77,268 KB
最終ジャッジ日時 2024-07-04 14:13:44
合計ジャッジ時間 3,586 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import defaultdict, deque
from heapq import heapify, heappop, heappush

n, m, s, g = map(int, input().split())
G = defaultdict(list)
for _ in range(m):
    a, b, c = map(int, input().split())
    G[a].append((b, c))
    G[b].append((a, c))
for i in range(n):
    G[i].sort(key=lambda x: x[0])
INF = 10**18
DP = [INF for _ in range(n)]
From = [-1 for _ in range(n)]
DP[g] = 0
H = [(0, g)]
while H:
    cc, cp = heappop(H)
    if cc > DP[cp]:
        continue
    for np, dc in G[cp]:
        nc = cc + dc
        if nc < DP[np]:
            DP[np] = nc
            From[np] = cp
            heappush(H, (nc, np))
        elif nc == DP[np]:
            From[np] = min(From[np], cp)

ANS = [s]
while ANS[-1] != g:
    cp = From[ANS[-1]]
    ANS.append(cp)
print(*ANS)
0