結果

問題 No.15 カタログショッピング
ユーザー Navier_BoltzmannNavier_Boltzmann
提出日時 2023-06-08 09:06:59
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 222 ms / 5,000 ms
コード長 1,307 bytes
コンパイル時間 312 ms
コンパイル使用メモリ 87,324 KB
実行使用メモリ 123,208 KB
最終ジャッジ日時 2023-08-29 00:05:19
合計ジャッジ時間 2,648 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 88 ms
73,088 KB
testcase_01 AC 88 ms
73,092 KB
testcase_02 AC 87 ms
72,892 KB
testcase_03 AC 92 ms
78,340 KB
testcase_04 AC 86 ms
72,968 KB
testcase_05 AC 209 ms
123,208 KB
testcase_06 AC 219 ms
120,380 KB
testcase_07 AC 221 ms
120,716 KB
testcase_08 AC 222 ms
120,720 KB
testcase_09 AC 217 ms
120,340 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