結果

問題 No.1457 ツブ消ししとるなHard
ユーザー 👑 KazunKazun
提出日時 2021-06-14 03:20:24
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 110 ms / 2,000 ms
コード長 864 bytes
コンパイル時間 236 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 84,840 KB
最終ジャッジ日時 2024-12-23 19:07:28
合計ジャッジ時間 1,948 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

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