結果

問題 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)
    


0