結果
問題 |
No.807 umg tours
|
ユーザー |
![]() |
提出日時 | 2021-12-05 12:36:29 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,154 ms / 4,000 ms |
コード長 | 1,100 bytes |
コンパイル時間 | 449 ms |
コンパイル使用メモリ | 82,052 KB |
実行使用メモリ | 171,640 KB |
最終ジャッジ日時 | 2024-07-07 07:35:03 |
合計ジャッジ時間 | 24,590 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 26 |
ソースコード
import heapq N,M = map(int,input().split()) ABC = [list(map(int,input().split())) for _ in range(M)] E = [[] for _ in range(N)] for a,b,c in ABC: E[a-1].append([b-1,c]) E[b-1].append([a-1,c]) D = [float('inf')] * N D2 = [float('inf')] * N #普通のダイクストラ def dijkstra(): D[0] = 0 q = [] heapq.heappush(q, [0, 0]) while len(q) > 0: d, u = heapq.heappop(q) if d > D[u]: continue for i in E[u]: a,b = i if D[a] > D[u] + b: D[a] = D[u] + b heapq.heappush(q, [D[a], a]) return #コスト-最大値の最小を取っていくダイクストラ def dijkstra2(): D2[0] = 0 q = [] heapq.heappush(q, [0, 0]) #コスト、頂点 while len(q) > 0: d, u = heapq.heappop(q) if d > D[u]: continue for i in E[u]: a,b = i tmp = min(D[u],D2[u] + b) if D2[a] > tmp: D2[a] = tmp heapq.heappush(q, [D2[a], a]) return dijkstra() dijkstra2() for i in range(N): print(D[i] + D2[i])