結果
問題 | No.160 最短経路のうち辞書順最小 |
ユーザー |
|
提出日時 | 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 |
ソースコード
from collections import defaultdict, dequefrom heapq import heapify, heappop, heappushn, 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**18DP = [INF for _ in range(n)]From = [-1 for _ in range(n)]DP[g] = 0H = [(0, g)]while H:cc, cp = heappop(H)if cc > DP[cp]:continuefor np, dc in G[cp]:nc = cc + dcif nc < DP[np]:DP[np] = ncFrom[np] = cpheappush(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)