結果

問題 No.1344 Typical Shortest Path Sum
ユーザー GER_chen
提出日時 2021-01-26 12:39:37
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 118 ms / 2,000 ms
コード長 817 bytes
コンパイル時間 179 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 76,800 KB
最終ジャッジ日時 2024-06-23 12:30:51
合計ジャッジ時間 7,607 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 77
権限があれば一括ダウンロードができます

ソースコード

diff # 

def floyd_warshall(n, Edges):
    """
    n: number of vertices
    Edges: set of edges
    """
    sp = [[float('inf')]*n for _ in range(n)]
    for init, end, weight in Edges:
        if sp[init][end] > weight:
            sp[init][end] = weight  #no multi edge
    for i in range(n):
        sp[i][i] = 0
    for i in range(n):
        for j in range(n):
            for k in range(n):
                if sp[j][k] > sp[j][i]+sp[i][k]:
                    sp[j][k] = sp[j][i]+sp[i][k]
    return sp

N, M = map(int, input().split())  #N頂点、M辺

#辺集合
Edges = set()
for _ in range(M):
    s, t, d = map(int, input().split())
    Edges.add((s-1, t-1, d))

#到達不可能な場合0にする設定
sp = floyd_warshall(N, Edges)
for i in range(N):
    print(sum(l if l < float('inf') else 0 for l in sp[i]))
0