結果
問題 | No.1344 Typical Shortest Path Sum |
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
def floyd_warshall(n, Edges):"""n: number of verticesEdges: 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 edgefor i in range(n):sp[i][i] = 0for 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 spN, 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]))