結果

問題 No.1457 ツブ消ししとるなHard
コンテスト
ユーザー 👑 Kazun
提出日時 2021-06-14 03:20:24
言語 PyPy3
(8.0.0 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 87 ms / 2,000 ms
+ 550µs
コード長 864 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 92 ms
コンパイル使用メモリ 64,388 KB
実行使用メモリ 73,272 KB
最終ジャッジ日時 2026-10-17 01:20:48
合計ジャッジ時間 2,577 ms
ジャッジサーバーID
(参考情報)
judge2_0 / judge4_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

from collections import defaultdict

N,M,X,Y,Z=map(int,input().split())
A=list(map(int,input().split()))

delete_like=delete_count=0
hold_like=hold_count=0

B=[]
for a in A:
    if a<=Y:
        delete_like+=a
        delete_count+=1
    elif a>=X:
        hold_like+=a
        hold_count+=1
    else:
        B.append(a-Z)

if hold_count>M:
    exit(print("Handicapped"))


if hold_count==N:
    exit(print(1 if sum(A)==N*Z else 0))

DP=[[defaultdict(int) for _ in range(len(B)+2)] for _ in range(len(B)+1)]
DP[0][0][0]=1

for i,b in enumerate(B,1):
    D=DP[i]; E=DP[i-1]
    for k in range(len(B)+1):
        e=E[k]
        d1=D[k]
        d2=D[k+1]
        for s in DP[i-1][k]:
            d1[s]+=e[s]
            d2[s+b]+=e[s]

U=hold_like-hold_count*Z
Ans=0
for i in range(min(len(B),M-hold_count)+1):
    Ans+=DP[-1][i][-U]

if U==0:
    Ans-=1

print(Ans)
0