結果

問題 No.807 umg tours
ユーザー silversmith
提出日時 2019-04-05 06:08:39
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,083 bytes
コンパイル時間 363 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 1,513,944 KB
最終ジャッジ日時 2024-12-30 13:11:37
合計ジャッジ時間 83,238 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 4 TLE * 8 MLE * 14
権限があれば一括ダウンロードができます

ソースコード

diff #

import os
import sys
import heapq

N, M = map(int, input().split(' '))
e={}
for i in range(N):
    for j in range(N):
        e[(i,j)] = sys.maxsize

for i in range(M):
    v1, v2, cost = [int(x) for x in input().split(' ')]
    e[(v1 -1, v2 -1)] = cost
    e[(v2 -1, v1 -1)] = cost

d = [sys.maxsize] * N
d_exe = [sys.maxsize] * N
d[0] = 0
d_exe[0] = 0
h = []
for i in range(N):
    heapq.heappush(h, (d[i],i))
while len(h) > 0:
    p = heapq.heappop(h)
    for idx, (_, i) in enumerate(h):
        if e[(p[1], i)] < sys.maxsize:
            alt = p[0] + e[(p[1], i)]
            if d[i] < alt : continue
            d[i] = alt
            h[idx] = (alt, i)
    heapq.heapify(h)

h = []    
for i in range(N):
    heapq.heappush(h, (d_exe[i],i))
while len(h) > 0:
    p = heapq.heappop(h)
    for idx, (_, i) in enumerate(h):
        if e[(p[1], i)] < sys.maxsize:
            alt = min(p[0] + e[(p[1], i)], d[p[1]])
            if d_exe[i] < alt : continue
            d_exe[i] = alt
            h[idx] = (alt, i)
    heapq.heapify(h)

for i in range(N):
    print(d[i] + d_exe[i])
0