結果
問題 |
No.2215 Slide Subset Sum
|
ユーザー |
![]() |
提出日時 | 2023-02-11 02:36:55 |
言語 | PyPy3 (7.3.15) |
結果 |
MLE
|
実行時間 | - |
コード長 | 989 bytes |
コンパイル時間 | 191 ms |
コンパイル使用メモリ | 82,412 KB |
実行使用メモリ | 664,380 KB |
最終ジャッジ日時 | 2024-07-07 19:28:46 |
合計ジャッジ時間 | 38,523 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 40 MLE * 5 |
ソースコード
mod = 998244353 def merge(X, Y, K): res = X[0] * Y[0] - 1 for i in range(1, K): res += X[i] * Y[K - i] res %= mod return res def calc_dp(K, A): dp = [0] * K dp[0] = 1 for a in A: tmp = [0] * K for i in range(K): tmp[i] = dp[i - a] for i in range(K): dp[i] += tmp[i] dp[i] %= mod yield dp[:] def main(): N, M, K = map(int, input().split()) block = (N + M - 1) // M A = [0] * (block * M + M) for i, a in enumerate(map(int, input().split())): A[i] = a answer = [] for i in range(block): L = [A[j] for j in range(M*i, M*i+M)][::-1] R = [A[j] for j in range(M*i+M, M*i+2*M)] dpL = list(calc_dp(K, L))[::-1] dpR = list(calc_dp(K, R)) answer.append((dpL[0][0] - 1) % mod) for j in range(M - 1): answer.append(merge(dpL[j+1], dpR[j], K)) print(*answer[:N - M + 1], sep="\n") main()