結果
問題 |
No.1238 選抜クラス
|
ユーザー |
![]() |
提出日時 | 2020-10-17 09:25:47 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 655 bytes |
コンパイル時間 | 239 ms |
コンパイル使用メモリ | 82,368 KB |
実行使用メモリ | 359,616 KB |
最終ジャッジ日時 | 2024-07-21 01:50:44 |
合計ジャッジ時間 | 8,925 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 TLE * 14 |
ソースコード
import sys from collections import defaultdict def MI(): return map(int,sys.stdin.readline().rstrip().split()) def LI(): return list(map(int,sys.stdin.readline().rstrip().split())) N,K = MI() A = LI() for i in range(N): A[i] -= K mod = 10**9+7 A = [0]+A # 和が0以上になる選び方を数え上げる dp = defaultdict(int) # dp[(i,j)] = 生徒1~iの中から選ぶ時、和がjになるような選び方 dp[(0,0)] = 1 for i in range(1,N+1): a = A[i] for j in range(-10000,10001): dp[(i,j)] = dp[(i-1,j)]+dp[(i-1,j-a)] dp[(i,j)] %= mod ans = -1 for i in range(10001): ans += dp[(N,i)] ans %= mod print(ans)