結果
問題 | No.2215 Slide Subset Sum |
ユーザー |
![]() |
提出日時 | 2023-02-10 22:55:58 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 583 ms / 3,000 ms |
コード長 | 914 bytes |
コンパイル時間 | 3,795 ms |
コンパイル使用メモリ | 254,772 KB |
最終ジャッジ日時 | 2025-02-10 13:28:15 |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 45 |
ソースコード
#include <stdio.h>#include <bits/stdc++.h>#include <atcoder/all>using namespace atcoder;using mint = modint998244353;using namespace std;#define rep(i,n) for (int i = 0; i < (n); ++i)#define Inf32 1000000001#define Inf64 1000000000000000001int main(){int N,M,K;cin>>N>>M>>K;vector<int> a(N);rep(i,N)cin>>a[i];for(int i=M-1;i<N;i+=M+1){vector ldp(M+1,vector<mint>(K,0));ldp[0][0] = 1;rep(j,M){int A = a[i-j];rep(k,K){ldp[j+1][k] += ldp[j][k];ldp[j+1][(k+A)%K] += ldp[j][k];}}vector<mint> rdp(K,0);rdp[0] = 1;int cur = i+1;while(ldp.size()>0){mint ans = 0;rep(j,K){ans += ldp.back()[j] * rdp[(K-j)%K];}cout<<(ans-1).val()<<endl;ldp.pop_back();if(cur>=N)break;vector<mint> ndp(K,0);rep(j,K){ndp[j] += rdp[j];ndp[(j+a[cur])%K] += rdp[j];}swap(ndp,rdp);cur++;}}return 0;}