結果

問題 No.807 umg tours
ユーザー norioc
提出日時 2024-09-03 03:01:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 3,197 ms / 4,000 ms
コード長 875 bytes
コンパイル時間 654 ms
コンパイル使用メモリ 82,292 KB
実行使用メモリ 173,664 KB
最終ジャッジ日時 2024-12-15 19:39:27
合計ジャッジ時間 30,620 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
from heapq import heappush, heappop

INF = 1 << 60
N, M = map(int, input().split())
adj = defaultdict(list)
for _ in range(M):
    a, b, c = map(int, input().split())
    a -= 1
    b -= 1
    adj[a].append((b, c))
    adj[b].append((a, c))

dists = [[INF] * 2 for _ in range(N)]
q = [(0, 0, 0)]  # (距離, チケット使用回数, 頂点)
while q:
    d, cnt, v = heappop(q)
    if dists[v][cnt] <= d: continue
    dists[v][cnt] = d

    # チケットを使う
    if cnt == 0:
        for to, _ in adj[v]:
            if dists[to][1] <= d: continue
            heappush(q, (d, 1, to))

    # チケットを使わない
    for to, cost in adj[v]:
        if dists[to][cnt] <= d+cost: continue
        heappush(q, (d+cost, cnt, to))

dists[0][0] = dists[0][1] = 0
for i in range(N):
    ans = dists[i][0] + dists[i][1]
    print(ans)
0