結果

問題 No.15 カタログショッピング
ユーザー Navier_BoltzmannNavier_Boltzmann
提出日時 2023-06-08 09:06:59
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 213 ms / 5,000 ms
コード長 1,307 bytes
コンパイル時間 415 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 120,808 KB
最終ジャッジ日時 2024-06-09 18:33:21
合計ジャッジ時間 2,037 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 43 ms
56,320 KB
testcase_01 AC 43 ms
55,424 KB
testcase_02 AC 44 ms
56,320 KB
testcase_03 AC 51 ms
64,384 KB
testcase_04 AC 43 ms
56,064 KB
testcase_05 AC 174 ms
120,808 KB
testcase_06 AC 186 ms
117,860 KB
testcase_07 AC 183 ms
118,136 KB
testcase_08 AC 213 ms
117,900 KB
testcase_09 AC 207 ms
117,748 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import *
from itertools import *
from functools import *
from heapq import *
import sys,math
input = sys.stdin.readline

N,S = map(int,input().split())
P = [int(input()) for i in range(N)]
if N==1:
    print(1)
    exit()

n = N//2
m = N-n
P0 = P[:n]
P1 = P[n:]
S0 = defaultdict(lambda:[])
S1 = defaultdict(lambda:[])
Z0 = []
Z1 = []
def cle(a,D):
    
    y = len(D)-1
    x = 0
    if D[x]>a:
        return 0
        
    if D[y]<=a:
        return y+1
    
    while y-x>1:
        mid = (y+x)//2
        if D[mid]<=a:
            x = mid
        else:
            y = mid
    return y
for i in range(1<<n):
    tmp = 0
    for j in range(n):
        if (i>>j)&1:
           tmp += P0[j]
    Z0.append(int(tmp))
    S0[tmp].append(i)

for i in range(1<<m):
    tmp = 0
    for j in range(m):
        if (i>>j)&1:
           tmp += P1[j]
           
    Z1.append(int(tmp))
    S1[tmp].append(i)

Z0 = list(set(Z0))
Z1 = list(set(Z1))
Z0.sort()
Z1.sort()

ans = []

for z0 in Z0:
    
    t = cle(S-z0,Z1)
    if t==0:
        continue
    for tx in S0[z0]:
        for ty in S1[S-z0]:
            ans.append(tx + (ty<<n))

Z = []
for a in ans:
    tmp = []
    for i in range(N):
        if (a>>i)&1:
          tmp.append(i+1)
    Z.append(tuple(tmp))
Z.sort()
for z in Z:
    print(*z)
0