結果

問題 No.15 カタログショッピング
ユーザー titiatitia
提出日時 2021-11-04 05:32:34
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 460 ms / 5,000 ms
コード長 797 bytes
コンパイル時間 397 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 24,180 KB
最終ジャッジ日時 2024-04-22 01:49:55
合計ジャッジ時間 3,389 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 34 ms
11,008 KB
testcase_01 AC 31 ms
10,752 KB
testcase_02 AC 31 ms
10,880 KB
testcase_03 AC 32 ms
10,752 KB
testcase_04 AC 31 ms
10,880 KB
testcase_05 AC 456 ms
24,052 KB
testcase_06 AC 459 ms
24,056 KB
testcase_07 AC 449 ms
23,980 KB
testcase_08 AC 454 ms
24,180 KB
testcase_09 AC 460 ms
24,056 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,S=map(int,input().split())
P=[int(input()) for i in range(N)]

Q=P[:N//2]
R=P[N//2:]

L=N//2
DP0=[0]*(1<<L)

L0=L

for i in range(1<<L):
    X=0
    for j in range(L):
        if i & (1<<j) != 0:
            X+=Q[j]
    DP0[i]=X

L=N-N//2
DP1=dict()

for i in range(1<<L):
    X=0
    for j in range(L):
        if i & (1<<j) != 0:
            X+=R[j]
    if X in DP1:
        DP1[X].append(i)
    else:
        DP1[X]=[i]

ANS=[]
for i in range(1<<L0):
    if S-DP0[i] in DP1:
        k=bin(i)[2:].zfill(L0)

        for j in DP1[S-DP0[i]]:
            l=bin(j)[2:].zfill(L)

            s=k[::-1]+l[::-1]
            X=[]
            for l in range(N):
                if s[l]=="1":
                    X.append(l+1)
            ANS.append(X)

for ans in sorted(ANS):
    print(*ans)
0