結果
問題 |
No.2215 Slide Subset Sum
|
ユーザー |
![]() |
提出日時 | 2023-02-10 22:49:03 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 884 bytes |
コンパイル時間 | 177 ms |
コンパイル使用メモリ | 82,144 KB |
実行使用メモリ | 111,924 KB |
最終ジャッジ日時 | 2024-07-07 16:53:12 |
合計ジャッジ時間 | 5,938 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | -- * 2 |
other | AC * 2 TLE * 1 -- * 42 |
ソースコード
mod = 998244353 div2 = pow(2, mod-2, mod) def calc_ans(dp, k): ans = 0 for i in range(0, len(dp), k): ans += dp[i] ans % mod return (ans - 1) % mod def main(): N, M, K = map(int, input().split()) A = list(map(lambda x: int(x) % K, input().split())) U = (K - 1) * M dp = [0] * (U + 1) dp[0] = 1 for i in range(M): for j in range(A[i], U+1)[::-1]: dp[j] += dp[j-A[i]] dp[j] %= mod print(calc_ans(dp, K)) for i in range(M, N): for j in range(A[i], U+1)[::-1]: dp[j] += dp[j-A[i]] dp[j] %= mod if A[i-M] != 0: for j in range(A[i-M], U+1): dp[j] -= dp[j-A[i-M]] dp[j] %= mod else: for j in range(U + 1): dp[j] = dp[j] * div2 % mod print(calc_ans(dp, K)) main()