結果

問題 No.1690 Power Grid
ユーザー roaris
提出日時 2021-09-24 23:01:53
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,022 bytes
コンパイル時間 249 ms
コンパイル使用メモリ 82,340 KB
実行使用メモリ 104,356 KB
最終ジャッジ日時 2024-07-05 11:10:46
合計ジャッジ時間 14,971 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 8 TLE * 1 -- * 16
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = sys.stdin.readline
from collections import *

N, M, K = map(int, input().split())
A = list(map(int, input().split()))
d = [[10**18]*N for _ in range(N)]

for i in range(N):
    d[i][i] = 0

for _ in range(M):
    X, Y, Z = map(int, input().split())
    d[X-1][Y-1] = Z
    d[Y-1][X-1] = Z

for k in range(N):
    for i in range(N):
        for j in range(N):
            d[i][j] = min(d[i][j], d[i][k]+d[k][j])

dp = [[10**18]*(1<<N) for _ in range(K+1)]
dp[0][0] = 0

for i in range(K):
    for S in range(1<<N):
        l1, l2 = [], []
        
        for v in range(N):
            if (S>>v)&1:
                l1.append(v)
            else:
                l2.append(v)
                
        for v in l2:
            if S==0:
                cost = 0
            else:
                cost = 10**18
            
                for nv in l1:
                    cost = min(cost, d[v][nv])
            
            dp[i+1][S|(1<<v)] = min(dp[i+1][S|(1<<v)], dp[i][S]+A[v]+cost)

print(min(dp[K]))
0