結果

問題 No.1344 Typical Shortest Path Sum
コンテスト
ユーザー GER_chen
提出日時 2021-01-26 11:43:20
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 778 bytes
コンパイル時間 377 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 76,928 KB
最終ジャッジ日時 2024-06-23 11:24:46
合計ジャッジ時間 6,917 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 32 WA * 45
権限があれば一括ダウンロードができます

ソースコード

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:
        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[i][j] > sp[i][k]+sp[k][j]:
                    sp[i][j] = sp[i][k]+sp[k][j]
    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