結果

問題 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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])
0