結果
問題 |
No.1238 選抜クラス
|
ユーザー |
|
提出日時 | 2020-11-18 16:18:20 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 112 ms / 2,000 ms |
コード長 | 428 bytes |
コンパイル時間 | 147 ms |
コンパイル使用メモリ | 82,196 KB |
実行使用メモリ | 76,856 KB |
最終ジャッジ日時 | 2024-07-23 09:04:30 |
合計ジャッジ時間 | 3,524 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 36 |
ソースコード
from collections import * N, K = map(int, input().split()) A = list(map(int, input().split())) A = list(map(lambda x: x-K, A)) dp = defaultdict(int) dp[0] = 1 MOD = 10**9+7 for i in range(N): ndp = defaultdict(int) for k in dp: ndp[k] = (ndp[k]+dp[k])%MOD ndp[k+A[i]] = (ndp[k+A[i]]+dp[k])%MOD dp = ndp ans = 0 for k in dp: if k>=0: ans = (ans+dp[k])%MOD print((ans-1)%MOD)