結果
問題 | No.1344 Typical Shortest Path Sum |
ユーザー |
![]() |
提出日時 | 2021-05-01 17:19:05 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 122 ms / 2,000 ms |
コード長 | 1,940 bytes |
コンパイル時間 | 201 ms |
コンパイル使用メモリ | 82,720 KB |
実行使用メモリ | 77,056 KB |
最終ジャッジ日時 | 2024-07-20 01:18:05 |
合計ジャッジ時間 | 7,022 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 77 |
ソースコード
class WarshallFloyd(): def __init__(self, N): self.N = N self.d = [[float("inf") for i in range(N)] for i in range(N)] # d[u][v] : 辺uvのコスト(存在しないときはinf) def add(self, u, v, c, directed=False): """ 0-indexedであることに注意 u = from, v = to, c = cost directed = Trueなら、有向グラフである """ #多重辺を持つ可能性がある #元のアルゴリズムは上書きとなっているが、これを更新に変更する必要がある if directed is False: self.d[u][v] = min(c,self.d[u][v]) self.d[v][u] = min(c,self.d[v][u]) else: self.d[u][v] = min(c,self.d[u][v]) def WarshallFloyd_search(self): # これを d[i][j]: iからjへの最短距離 にする # 本来無向グラフでのみ全域木を考えるが、二重辺なら有向でも行けそう # d[i][i] < 0 なら、グラフは負のサイクルを持つ for k in range(self.N): for i in range(self.N): for j in range(self.N): self.d[i][j] = min( self.d[i][j], self.d[i][k] + self.d[k][j]) hasNegativeCycle = False for i in range(self.N): if self.d[i][i] < 0: hasNegativeCycle = True break for i in range(self.N): self.d[i][i] = 0 return hasNegativeCycle, self.d N, M = map(int, input().split()) STD = [list(map(int, input().split())) for i in range(M)] graph = WarshallFloyd(N) for s,t,d in STD: graph.add(s-1, t-1, d, True) hasNegativeCycle, d = graph.WarshallFloyd_search()#負経路はないため、使用するのはdの方のみ for i in range(N): for j in range(N): if d[i][j] == float("inf"): d[i][j]=0 for i in range(N): print(sum(d[i]))