結果

問題 No.160 最短経路のうち辞書順最小
ユーザー norioc
提出日時 2024-07-18 23:28:50
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 89 ms / 5,000 ms
コード長 755 bytes
コンパイル時間 254 ms
コンパイル使用メモリ 82,152 KB
実行使用メモリ 77,492 KB
最終ジャッジ日時 2024-07-18 23:28:55
合計ジャッジ時間 3,875 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import defaultdict, deque

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

dists = [INF] * N
q = deque([(0, G)])  # ゴールから探索する
while q:
    d, v = q.popleft()
    if dists[v] <= d: continue
    dists[v] = d

    for to, cost in adj[v]:
        if dists[to] <= d + cost: continue
        q.append((d+cost, to))

# 経路復元
path = [S]  # スタート地点から復元
while (v := path[-1]) != G:
    nv = INF  # 次の頂点
    for to, cost in adj[v]:
        if dists[to]+cost == dists[v]:
            nv = min(nv, to)

    assert nv != INF
    path.append(nv)

print(*path)
0