結果
問題 |
No.1555 Constructed Balancing Sequence
|
ユーザー |
![]() |
提出日時 | 2024-08-07 14:15:39 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 21 ms / 2,000 ms |
コード長 | 1,446 bytes |
コンパイル時間 | 1,174 ms |
コンパイル使用メモリ | 109,384 KB |
実行使用メモリ | 9,472 KB |
最終ジャッジ日時 | 2024-08-07 14:15:43 |
合計ジャッジ時間 | 4,264 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 42 |
ソースコード
#include<iostream> #include<vector> #include<algorithm> using namespace std; using ll = long long; #include<atcoder/modint> using mint = atcoder::modint998244353; int main(){ cin.tie(nullptr); ios::sync_with_stdio(false); ll n,k; cin>>n>>k; vector<ll> a(n); for(int i = 0;i<n;i++) cin>>a[i]; ll sum = 0; for(int i = 0;i<n;i++) sum += a[i]; vector<vector<mint>> dp(n+1,vector<mint>(3,0)); dp[n][1] = 1; for(int i = n-1;i>=0;i--){ sum -= a[i]; if(sum==a[i]) { mint b = mint(a[i]-(-k)); dp[i][2] += b * dp[i+1][1]; dp[i][0] += dp[i+1][1]; dp[i][0] += mint(a[i]-(-k)+1) * dp[i+1][0]; dp[i][2] += mint(a[i]-(-k)+1) * dp[i+1][2]; }else if(sum==a[i]+1){ mint b = mint(a[i]-(-k)); dp[i][2] += b * dp[i+1][1]; dp[i][1] += dp[i+1][1]; dp[i][0] += mint(a[i]-(-k)+1) * dp[i+1][0]; dp[i][2] += mint(a[i]-(-k)+1) * dp[i+1][2]; }else{ mint b = mint(a[i]-(-k)); dp[i][2] += b * dp[i+1][1]; dp[i][1] += dp[i+1][1]; dp[i][1] += mint(a[i]-(-k)+1) * dp[i+1][0]; dp[i][2] += mint(a[i]-(-k)+1) * dp[i+1][2]; } } mint ans = 1; for(int i = 0;i<n;i++) ans *= (a[i]-(-k)+1); ans -= dp[0][2]; cout<<ans.val()<<endl; ans = dp[0][0] + dp[0][1]; //cout<<ans.val()<<endl; }