結果

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

ソースコード

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(str(i + 1))

    l = " ".join(l)

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

      r = " ".join(r)

      if min(len(l), len(r)) == 0:
        ans.append(l + r)
      else:
        ans.append(l + " " + r)

ans.sort()

print(*ans, sep="\n")
0