結果

問題 No.15 カタログショッピング
コンテスト
ユーザー qib
提出日時 2023-03-18 14:33:40
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 119 ms / 5,000 ms
コード長 998 bytes
コンパイル時間 191 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 91,904 KB
最終ジャッジ日時 2024-09-18 13:18:00
合計ジャッジ時間 1,520 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

n, s = map(int, input().split())
p = [int(input()) for _ in range(n)]

left = {}
for mask in range(1 << (n // 2)):
  cost = 0
  for i in range(n // 2):
    if (mask >> i) & 1 != 0:
      cost += p[i]

  if cost > s:
    continue

  if left.get(cost, None) is None:
    left[cost] = []

  left[cost].append(mask)

right = {}
for mask in range(1 << (n - n // 2)):
  cost = 0
  for i in range(n // 2, n):
    if (mask >> (i - n // 2)) & 1 != 0:
      cost += p[i]

  if cost > s:
    continue

  if right.get(cost, None) is None:
    right[cost] = []

  right[cost].append(mask)

ans = []
for k, ls in left.items():
  if right.get(s - k, None) is None:
    continue

  for lm in ls:
    l = []
    for i in range(n // 2):
      if (lm >> i) & 1 != 0:
        l.append(i + 1)

    for rm in right[s - k]:
      r = []
      for i in range(n - n // 2):
        if (rm >> i) & 1 != 0:
          r.append(i + 1 + n // 2)

      ans.append(l + r)

ans.sort()

for xs in ans:
  print(" ".join(map(str, xs)))
0