結果

問題 No.1929 Exponential Sequence
ユーザー とりゐとりゐ
提出日時 2022-04-04 23:47:09
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 325 ms / 2,000 ms
コード長 736 bytes
コンパイル時間 314 ms
コンパイル使用メモリ 87,112 KB
実行使用メモリ 142,508 KB
最終ジャッジ日時 2023-10-12 04:18:54
合計ジャッジ時間 4,889 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 69 ms
71,620 KB
testcase_01 AC 68 ms
71,268 KB
testcase_02 AC 68 ms
71,300 KB
testcase_03 AC 325 ms
142,508 KB
testcase_04 AC 67 ms
71,128 KB
testcase_05 AC 68 ms
71,124 KB
testcase_06 AC 68 ms
71,640 KB
testcase_07 AC 69 ms
71,308 KB
testcase_08 AC 68 ms
71,236 KB
testcase_09 AC 68 ms
71,516 KB
testcase_10 AC 68 ms
71,284 KB
testcase_11 AC 68 ms
71,232 KB
testcase_12 AC 68 ms
71,548 KB
testcase_13 AC 67 ms
71,384 KB
testcase_14 AC 68 ms
71,312 KB
testcase_15 AC 80 ms
76,492 KB
testcase_16 AC 70 ms
71,468 KB
testcase_17 AC 67 ms
71,504 KB
testcase_18 AC 67 ms
71,480 KB
testcase_19 AC 68 ms
71,416 KB
testcase_20 AC 67 ms
71,548 KB
testcase_21 AC 319 ms
142,048 KB
testcase_22 AC 244 ms
117,120 KB
testcase_23 AC 317 ms
141,780 KB
testcase_24 AC 197 ms
125,640 KB
testcase_25 AC 311 ms
136,576 KB
testcase_26 AC 319 ms
142,200 KB
testcase_27 AC 69 ms
71,444 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

n,s=map(int,input().split())
a=list(map(int,input().split()))

def calc(m,b):
  dp=[0]
  for i in b:
    dp_new=[]
    tmp=1
    while i**tmp<=s:
      for j in dp:
        if j+i**tmp<=s:
          dp_new.append(j+i**tmp)
      tmp+=1
    dp=dp_new
  dp.sort()
  dp2=[]
  for i in dp:
    if len(dp2)==0:
      dp2.append([i,1])
    elif dp2[-1][0]==i:
      dp2[-1][1]+=1
    else:
      dp2.append([i,1])
  
  cum=[]
  tmp=0
  for i,j in dp2:
    tmp+=j
    cum.append([i,tmp])
  return dp2,cum

n1=n//2
n2=n-n1
dp,_=calc(n1,a[:n1])
_,cum=calc(n2,a[n1:])
cum=cum[::-1]
m1=len(dp)
m2=len(cum)
ans=0
now=0
for i in range(m1):
  while now<m2 and dp[i][0]+cum[now][0]>s:
    now+=1
  if now!=m2:
    ans+=dp[i][1]*cum[now][1]

print(ans)
0