結果

問題 No.15 カタログショッピング
ユーザー dangodango
提出日時 2023-07-08 17:50:30
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 230 ms / 5,000 ms
コード長 904 bytes
コンパイル時間 205 ms
コンパイル使用メモリ 82,204 KB
実行使用メモリ 90,252 KB
最終ジャッジ日時 2024-07-22 13:17:44
合計ジャッジ時間 2,329 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
52,872 KB
testcase_01 AC 37 ms
53,028 KB
testcase_02 AC 39 ms
53,080 KB
testcase_03 AC 47 ms
61,856 KB
testcase_04 AC 38 ms
52,772 KB
testcase_05 AC 225 ms
90,184 KB
testcase_06 AC 230 ms
89,864 KB
testcase_07 AC 228 ms
90,252 KB
testcase_08 AC 223 ms
90,044 KB
testcase_09 AC 227 ms
89,996 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from itertools import combinations

def main():
    n, s = map(int, input().split())
    pi = []
    for _ in range(n):
      pi.append(int(input()))

    if n == 1:
        if pi[0] == s:
            print(1)
        else:
            print(-1)
        return

    m1 = n // 2
    m2 = n - m1

    hi = {}
    for i in range(1 << m2):
        u = sum(pi[j + m1] for j in range(m2) if (i >> j) & 1)
        if u not in hi:
            hi[u] = []
        hi[u].append(i)

    ri = []
    for i in range(1 << m1):
        u = sum(pi[j] for j in range(m1) if (i >> j) & 1)
        w = s - u
        if w < 0 or w not in hi:
            continue
        ri.extend([(x << m1) | i for x in hi[w]])

    if not ri:
        print(-1)
    else:
        ri2 = [sorted([j + 1 for j in range(n) if (r >> j) & 1]) for r in ri]
        ri2.sort()
        for r in ri2:
            print(" ".join(map(str, r)))

main()
0