結果

問題 No.1929 Exponential Sequence
ユーザー Shirotsume
提出日時 2022-05-06 22:20:39
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 206 ms / 2,000 ms
コード長 937 bytes
コンパイル時間 237 ms
コンパイル使用メモリ 81,960 KB
実行使用メモリ 84,472 KB
最終ジャッジ日時 2024-09-14 03:53:21
合計ジャッジ時間 3,318 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#
import sys
input = lambda: sys.stdin.readline().rstrip()
ii = lambda: int(input())
mi = lambda: map(int, input().split())
li = lambda: list(mi())
INF = 2 ** 63 - 1
def solve(a, s):
from collections import Counter
n = len(a)
dp = Counter()
dp[0] = 1
for i in range(n):
e = Counter()
for j in range(1, 40):
now = a[i] ** j
if now >= s:
break
for v, c in dp.items():
e[v + now] += c
dp = e
return dp
n, s = mi()
a = li()
a.sort()
a1 = a[:n // 2]
a2 = a[n//2:]
dp1 = list(solve(a1, s).items())
dp2 = list(solve(a2, s).items())
dp1.sort()
dp2.sort()
summ = 0
for v, c in dp2:
summ += c
now = len(dp2) - 1
ans = 0
for v, c in dp1:
while now >= 0:
if dp2[now][0] + v <= s:
break
summ -= dp2[now][1]
now -= 1
ans += c * summ
print(ans)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0