結果

問題 No.2869 yuusaan's Knapsacks
コンテスト
ユーザー 寝癖
提出日時 2024-07-20 12:03:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 3,564 ms / 4,500 ms
コード長 1,046 bytes
コンパイル時間 372 ms
コンパイル使用メモリ 82,524 KB
実行使用メモリ 210,908 KB
最終ジャッジ日時 2024-08-04 18:05:09
合計ジャッジ時間 41,283 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

N, M = map(int, input().split())
e = list(map(int, input().split()))
v, w = [], []
for _ in range(M):
    vi, wi = map(int, input().split())
    v.append(vi)
    w.append(wi)

K = 1<<M
sum_v, sum_w = [0]*K, [0]*K
for S in range(K):
    for i in range(M):
        if S>>i&1:
            sum_v[S] += v[i]
            sum_w[S] += w[i]

dp = [0]*(K*(N+1))
rev = [{} for _ in range(N+1)]
for i in range(N):
    for S in range(K):
        U = K - 1 - S
        T = U
        while True:
            if sum_w[T] <= e[i] and dp[(i+1)*K+(S|T)] < dp[i*K+S]+sum_v[T]:
                # (i, S) -> (i+1, S|T)
                dp[(i+1)*K+(S|T)] = dp[i*K+S]+sum_v[T]
                rev[i+1][S|T] = S
            if T == 0:
                break
            T = (T-1)&U

m = max(dp[N*K+S] for S in range(K))
S = 0
while dp[N*K+S] != m:
    S += 1

ans = [[] for _ in range(N)]
for i in range(N, 0, -1):
    T = rev[i].get(S, S)
    for j in range(M):
        if (S^T)>>j&1:
            ans[i-1].append(j+1)
    S = T

print(m)
for a in ans:
    print(len(a), *a)
0