結果
問題 | No.17 2つの地点に泊まりたい |
ユーザー |
|
提出日時 | 2015-08-05 23:43:04 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 867 bytes |
コンパイル時間 | 80 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 11,008 KB |
最終ジャッジ日時 | 2024-07-18 03:33:14 |
合計ジャッジ時間 | 2,181 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 18 WA * 9 |
ソースコード
def read_data(): N = int(input()) S = [int(input()) for n in range(N)] M = int(input()) dist = [[float('inf')] * N for n in range(N)] for n in range(N): dist[n][n] = 0 for m in range(M): A, B, C = map(int, input().split()) dist[A][B] = C dist[B][A] = C return N, S, dist def solve(N, S, dist): for i in range(N): dist_i = dist[i] for j in range(N): for k in range(N): dist[i][j] = min(dist_i[j], dist_i[k] + dist[k][j]) mincost = float('inf') for i in range(1, N-1): for j in range(1, N-1): if i == j: continue cost = dist[0][i] + dist[i][j] + dist[j][N-1] + S[i] + S[j] if cost < mincost: mincost = cost return mincost N, S, dist = read_data() print(solve(N, S, dist))