結果
問題 | No.160 最短経路のうち辞書順最小 |
ユーザー |
|
提出日時 | 2022-12-14 18:47:26 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 123 ms / 5,000 ms |
コード長 | 799 bytes |
コンパイル時間 | 458 ms |
コンパイル使用メモリ | 81,792 KB |
実行使用メモリ | 77,440 KB |
最終ジャッジ日時 | 2024-11-08 10:33:26 |
合計ジャッジ時間 | 3,937 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 26 |
ソースコード
import heapq INF = 1 << 60 n, m, src, dst = map(int, input().split()) g = [[] for _ in range(n)] for _ in range(m): a, b, c = map(int, input().split()) g[a].append((b, c)) g[b].append((a, c)) for u in range(n): g[u].sort(key=lambda x: x[0]) dist = [INF for _ in range(n)] dist[dst] = 0 hp = [(dist[dst], dst)] par = [None for _ in range(n)] while len(hp) > 0: cd, cur = heapq.heappop(hp) if cd > dist[cur]: continue for nxt, c in g[cur]: if dist[cur] + c < dist[nxt]: par[nxt] = cur dist[nxt] = dist[cur] + c heapq.heappush(hp, (dist[nxt], nxt)) cur = src ans = [] while True: ans.append(str(cur)) if cur == dst: break nxt = None for v, c in g[cur]: if dist[v] + c == dist[cur]: nxt = v break cur = nxt print(' '.join(ans))