結果

問題 No.160 最短経路のうち辞書順最小
ユーザー lloyzlloyz
提出日時 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)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0