結果
問題 | 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, 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)