結果

問題 No.1690 Power Grid
コンテスト
ユーザー puzneko
提出日時 2021-11-23 23:27:14
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 422 ms / 3,000 ms
コード長 1,156 bytes
コンパイル時間 308 ms
コンパイル使用メモリ 82,444 KB
実行使用メモリ 78,320 KB
最終ジャッジ日時 2024-06-25 21:14:12
合計ジャッジ時間 6,740 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from sys import stdin
from itertools import combinations

n, m, buildnum, *indata = map(int, stdin.read().split())
a = []
for i in range(n):
    a.append(indata[i])
offset = n
infty = 10**18
dist = [[infty for i in range(n)] for j in range(n)]
for i in range(m):
    s, t, d  = indata[offset + 3*i],indata[offset + 3*i+1],indata[offset + 3*i+2]
    dist[s-1][t-1] = d
    dist[t-1][s-1] = d

for i in range(n):
    for j in range(n):
        for k in range(n):
            dist[j][k] = min(dist[j][i]+dist[i][k], dist[j][k])

dp = [infty for i in range(1<<n)]
dp[0] = 0
for i in range(n):
    dp[(1<<i)] = a[i]
for i in range(2,buildnum+1):
    for p in combinations(range(n), (i)):
        ind = 0
        for j in p:
            ind = ind | (1<<j)
        for j in p:
            kari = ind ^ (1<<j)
            distmin = infty
            for k in p:
                if j != k:
                    distmin = min(distmin,dist[j][k])
            dp[ind] = min(dp[ind],dp[kari]+distmin+a[j])

ans = infty
for p in combinations(range(n), (buildnum)):
    ind = 0
    for j in p:
        ind = ind | (1<<j)
    ans = min(ans,dp[ind])

print("{}".format(ans))
0