結果
問題 |
No.1238 選抜クラス
|
ユーザー |
![]() |
提出日時 | 2020-10-17 09:27:11 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
TLE
|
実行時間 | - |
コード長 | 670 bytes |
コンパイル時間 | 87 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 273,240 KB |
最終ジャッジ日時 | 2024-07-21 01:51:05 |
合計ジャッジ時間 | 8,947 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 14 TLE * 1 -- * 21 |
ソースコード
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)