結果
| 問題 |
No.807 umg tours
|
| コンテスト | |
| ユーザー |
ntuda
|
| 提出日時 | 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])
ntuda