結果

問題 No.1344 Typical Shortest Path Sum
ユーザー GER_chen
提出日時 2021-01-26 12:32:00
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,006 bytes
コンパイル時間 153 ms
コンパイル使用メモリ 82,400 KB
実行使用メモリ 78,172 KB
最終ジャッジ日時 2024-06-23 12:25:43
合計ジャッジ時間 7,697 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other WA * 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[i][j] > sp[i][k]+sp[k][j]:
                    sp[i][j] = sp[i][k]+sp[k][j]
    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)
print(sp)
for i in range(N):
    print(sum(l if l < float('inf') else 0 for l in sp[i]))
0