結果

問題 No.2364 Knapsack Problem
ユーザー ryohei22ryohei22
提出日時 2023-06-30 22:29:03
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 104 ms / 3,000 ms
コード長 1,425 bytes
コンパイル時間 240 ms
コンパイル使用メモリ 82,172 KB
実行使用メモリ 78,424 KB
最終ジャッジ日時 2024-07-07 10:19:48
合計ジャッジ時間 2,359 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict


N, M, W = map(int, input().split())
A = list(map(int, input().split()))
B = list(map(int, input().split()))
A += list(map(lambda x: -int(x), input().split()))
B += list(map(lambda x: -int(x), input().split()))

# bitDP
# dp[i][j][k]: i回各操作を行うかを決め、行った操作の集合がjであり、重さがkである時の価値の最大値
# MLEでたから、dp[i-1]とdp[i]のみを、oldとnewwとして保持してみる
old = defaultdict(lambda: defaultdict(int))
old[0][0] = 0
for _ in range(N+M):
    neww = defaultdict(lambda: defaultdict(int))
    for j in old:
        for k in old[j]:
            if j not in neww or k not in neww[j]:
                neww[j][k] = old[j][k]
            else:
                neww[j][k] = max(neww[j][k], old[j][k])
            for l in range(N+M):
                # l番目の操作をまだ行っていない かつ 操作を行っても重さが条件を満たす
                if not j >> l & 1 and 0 <= k + A[l] <= W:
                    if j | 2**l not in neww or k + A[l] not in neww[j|2**l]:
                        neww[j|2**l][k+A[l]] = old[j][k] + B[l]
                    else:
                        neww[j|2**l][k+A[l]] \
                            = max(neww[j|2**l][k+A[l]], old[j][k] + B[l])
    old = neww

ans = -float('inf')
for j in old:
    for k in old[j]:
        ans = max(ans, old[j][k])
print(ans)
0