結果
問題 |
No.160 最短経路のうち辞書順最小
|
ユーザー |
![]() |
提出日時 | 2020-12-12 13:27:39 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 85 ms / 5,000 ms |
コード長 | 853 bytes |
コンパイル時間 | 160 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 77,088 KB |
最終ジャッジ日時 | 2024-09-19 21:55:54 |
合計ジャッジ時間 | 3,319 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 26 |
ソースコード
from heapq import * def dijkstra(g,start): n = len(g) INF = 1<<61 dist = [INF]*(n) #startからの最短距離 #pending = n-1 #未確定の点の個数 dist[start] = 0 q = [(0,start)] #(そこまでの距離、点) while q:# and pending: dv,v = heappop(q) if dist[v] < dv: continue for to, cost in g[v]: if dv + cost < dist[to]: dist[to] = dv + cost heappush(q, (dist[to], to)) return dist n,m,s,goal = 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)) dist = dijkstra(g,goal) res = [s] v = s while v != goal: i = n+1 for u,c in g[v]: if dist[v] == dist[u] + c: i = min(i,u) v = i res.append(v) print(*res)