結果

問題 No.17 2つの地点に泊まりたい
ユーザー OKCH3COOH
提出日時 2019-10-25 00:37:43
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 69 ms / 5,000 ms
コード長 758 bytes
コンパイル時間 187 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 11,008 KB
最終ジャッジ日時 2024-07-18 09:12:18
合計ジャッジ時間 2,005 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
from heapq import heappop, heappush

N = int(input())

stopCost = [int(input()) for _ in range(N)]

M = int(input())
minDist = [[float('inf')] * N for _ in range(N)]

for _ in range(M):
    fr, to, cost = map(int, input().split())
    minDist[fr][to] = cost
    minDist[to][fr] = cost

for k in range(N):
    for i in range(N):
        for j in range(N):
            d = minDist[i][k] + minDist[k][j]
            if minDist[i][j] > d:
                minDist[i][j] = d

ans = float('inf')
for i in range(1, N - 1):
    for j in range(i + 1, N - 1):
        dist = min(minDist[0][i] + minDist[j][N - 1], minDist[N - 1][i] + minDist[j][0]) + minDist[i][j]
        ans = min(ans, dist + stopCost[i] + stopCost[j])

print(ans)
0