結果

問題 No.160 最短経路のうち辞書順最小
ユーザー rlangevin
提出日時 2023-08-16 12:33:19
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 199 ms / 5,000 ms
コード長 813 bytes
コンパイル時間 152 ms
コンパイル使用メモリ 82,564 KB
実行使用メモリ 84,600 KB
最終ジャッジ日時 2024-11-25 01:23:10
合計ジャッジ時間 5,923 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import *

N, M, S, G = map(int, input().split())
inf = 10 ** 18
dp = [[inf] * N for i in range(N)]
for i in range(N):
    dp[i][i] = 0

Edge = defaultdict(lambda:inf)
for i in range(M):
    a, b, c = map(int, input().split())
    dp[a][b] = min(dp[a][b], c)
    dp[b][a] = min(dp[b][a], c)
    Edge[(a, b)] = min(Edge[(a, b)], c)
    Edge[(b, a)] = min(Edge[(b, a)], c)

for k in range(N):
    for i in range(N):
        for j in range(N):
            dp[i][j] = min(dp[i][j], dp[i][k] + dp[k][j])

ans = [S]
now = S
cost = dp[now][G]
while now != G:
    for i in range(N):
        if i == now:
            continue
        if dp[now][i] + dp[i][G] == cost and Edge[now, i] == dp[now][i]:
            ans.append(i)
            cost -= dp[now][i]
            now = i
            break

print(*ans)
0