結果

問題 No.943 取り調べ
ユーザー OKCH3COOH
提出日時 2019-12-08 18:49:51
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 835 bytes
コンパイル時間 280 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 224,580 KB
最終ジャッジ日時 2024-12-31 07:40:01
合計ジャッジ時間 23,709 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 14 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import defaultdict

N = int(input())
INF = 10**10

parents = [tuple(input().split()) for _ in range(N)]
A = list(map(int, input().split()))

masks = []
for mask in range(1 << N):
    V = set(i for i in range(N) if (mask & (1 << i)) != 0)
    masks.append(V)
masks.sort(key=lambda B: len(B))

dp = defaultdict(lambda : INF)
dp[()] = 0

for mask in masks:
    V = list(mask)
    V.sort()
    V = tuple(V)

    for d in range(N):
        if d in mask:
            continue

        cost = 0
        state = list(mask) + [d]
        for i, (s, a) in enumerate(zip(parents[d], A)):
            if s == '1' and (i not in mask):
                state.append(i)
                cost += a
        state.sort()
        state = tuple(state)
        dp[state] = min(dp[state], dp[V] + cost)

print(dp[tuple(i for i in range(N))])
0