結果

問題 No.2869 yuusaan's Knapsacks
ユーザー PNJ
提出日時 2024-09-02 10:58:31
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 383 ms / 4,500 ms
コード長 2,675 bytes
コンパイル時間 653 ms
コンパイル使用メモリ 82,104 KB
実行使用メモリ 109,240 KB
最終ジャッジ日時 2024-09-02 10:58:39
合計ジャッジ時間 7,919 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 67280421310721

def popcount(n):
  c=(n&0x5555555555555555)+((n>>1)&0x5555555555555555)
  c=(c&0x3333333333333333)+((c>>2)&0x3333333333333333)
  c=(c&0x0f0f0f0f0f0f0f0f)+((c>>4)&0x0f0f0f0f0f0f0f0f)
  c=(c&0x00ff00ff00ff00ff)+((c>>8)&0x00ff00ff00ff00ff)
  c=(c&0x0000ffff0000ffff)+((c>>16)&0x0000ffff0000ffff)
  c=(c&0x00000000ffffffff)+((c>>32)&0x00000000ffffffff)
  return c

def zeta_transform(f,upper = 0):
  n = (len(f) - 1).bit_length()
  if upper:
    for i in range(n):
      k = 1 << i
      for bit in range(1 << n):
        if bit & k == 0:
          f[bit] = (f[bit] + f[bit ^ k]) % mod
    return
  for i in range(n):
    k = 1 << i
    for bit in range(1 << n):
      if bit & k:
        f[bit] = (f[bit] + f[bit ^ k]) % mod
  return

def mobius_transform(f,upper = 0):
  n = (len(f) - 1).bit_length()
  if upper:
    for i in range(n):
      k = 1 << i
      for bit in range(1 << n):
        if bit & k == 0:
          f[bit] = (f[bit] - f[bit ^ k]) % mod
    return
  for i in range(n):
    k = 1 << i
    for bit in range(1 << n):
      if bit & k:
        f[bit] = (f[bit] - f[bit ^ k]) % mod
  return

def or_convolution(a,b):
  f = a[:]
  g = b[:]
  zeta_transform(f)
  zeta_transform(g)
  for i in range(len(f)):
    f[i] = f[i] * g[i] % mod
  mobius_transform(f)
  for i in range(len(f)):
    if f[i] > 0:
      f[i] = 1
  return f

def and_convolution(a,b):
  f = a[:]
  g = b[:]
  zeta_transform(f,1)
  zeta_transform(g,1)
  for i in range(len(f)):
    f[i] = f[i] * g[i] % mod
  mobius_transform(f,1)
  return f

N,M = map(int,input().split())
E = list(map(int,input().split()))
V,W = [0] * M,[0] * M
for i in range(M):
  V[i],W[i] = map(int,input().split())

SW = [0 for _ in range(1 << M)]
SV = [0 for _ in range(1 << M)]
for s in range(1 << M):
  w,v = 0,0
  for j in range(M):
    if (s >> j) & 1:
      w += W[j]
      v += V[j]
  SW[s] = w
  SV[s] = v

dp = [[0 for s in range(1 << M)] for n in range(N)]
for n in range(N):
  if n == 0:
    for s in range(1 << M):
      if SW[s] <= E[n]:
        dp[n][s] = 1
    continue
  B = [0 for s in range(1 << M)]
  for s in range(1 << M):
    if SW[s] <= E[n]:
      B[s] = 1
  dp[n] = or_convolution(dp[n - 1],B)

ans = 0
s = 0
for ss in range(1 << M):
  if dp[-1][ss]:
    if SV[ss] > ans:
      ans = SV[ss]
      s = ss

res = [0 for i in range(N)]
for n in range(N - 1,0,-1):
  for ss in range(1 << M):
    if (s & ss) != ss:
      continue
    if dp[n - 1][s ^ ss] & (SW[ss] <= E[n]):
      res[n] = ss
      s ^= ss
      break
res[0] = s

print(ans)
for n in range(N):
  X = []
  X.append(popcount(res[n]))
  for j in range(M):
    if (res[n] >> j) & 1:
      X.append(j + 1)
  print(*X)
0