結果

問題 No.2625 Bouns Ai
ユーザー zeta7532zeta7532
提出日時 2024-01-11 01:20:24
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 226 ms / 2,000 ms
コード長 937 bytes
コンパイル時間 2,410 ms
コンパイル使用メモリ 208,716 KB
実行使用メモリ 161,792 KB
最終ジャッジ日時 2024-09-27 20:23:30
合計ジャッジ時間 7,727 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
using namespace std;
using ll = long long;
const ll mod = 998244353;
#define fi first
#define se second
#define rep(i,n) for(ll i=0;i<n;i++)
#define all(x) x.begin(),x.end()
#define faster ios::sync_with_stdio(false);cin.tie(nullptr)

int main() {
    ll N;
    cin >> N;
    vector<ll> A(N);
    rep(i,N) cin >> A[i];
    vector<vector<ll>> dp(N,vector<ll>(100500,0));
    rep(j,100500) if(j>=A[0]) dp[0][j]=1;
    rep(i,N-1){
        vector<ll> cum(100500,0);
        cum[0]=dp[i][0];
        rep(j,100500-1) cum[j+1]=(cum[j]+dp[i][j+1])%mod;
        rep(j,100500){
            if(j<A[i+1]) continue;
            if(A[i]<A[i+1]) dp[i+1][j]=cum[j-A[i+1]+A[i]];
            if(A[i]>=A[i+1]) dp[i+1][j]=cum[j];
        }
    }
    ll ans=0;
    rep(j,100000+1) ans+=dp[N-1][j];
    cout << ans%mod << endl;
    return 0;
}
0