結果

問題 No.15 カタログショッピング
ユーザー qibqib
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 40 ms
52,352 KB
testcase_01 AC 44 ms
51,840 KB
testcase_02 AC 41 ms
52,864 KB
testcase_03 AC 46 ms
60,544 KB
testcase_04 AC 40 ms
52,096 KB
testcase_05 AC 119 ms
91,904 KB
testcase_06 AC 114 ms
91,392 KB
testcase_07 AC 117 ms
91,648 KB
testcase_08 AC 116 ms
90,408 KB
testcase_09 AC 118 ms
90,592 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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