結果

問題 No.2625 Bouns Ai
ユーザー maeshunmaeshun
提出日時 2024-02-09 22:33:55
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 88 ms / 2,000 ms
コード長 1,122 bytes
コンパイル時間 4,007 ms
コンパイル使用メモリ 265,268 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-09-28 15:46:31
合計ジャッジ時間 6,566 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
#define rep(i, n) for(int i=0;i<(n);++i)
#define rep1(i, n) for(int i=1;i<=(n);i++)
#define ll long long
using mint = modint998244353;
using P = pair<ll,ll>;
using lb = long double;
using T = tuple<ll, ll, ll>;
#ifdef LOCAL
#  include <debug_print.hpp>
#  define dbg(...) debug_print::multi_print(#__VA_ARGS__, __VA_ARGS__)
#else
#  define dbg(...) (static_cast<void>(0))
#endif

int main()
{
    int n;
    cin >> n;
    vector<int> a(n);
    rep(i,n) cin >> a[i];
    int m = 1e5;
    vector<mint> dp(m+1);
    rep(i,m+1){
        if(i+a[0]<=m) dp[i] = 1;
    }
    for(int i=1;i<n;i++){
        vector<mint> sum(m+2);
        rep(j,m+1) sum[j+1] = sum[j] + dp[j];
        vector<mint> p(m+1);
        swap(dp, p);
        rep(j,m+1){
            if(j+a[i]>m) break;
            int b = min(j + a[i] - a[i-1], j);
            if(b<0) continue;
            dp[j] += sum[b+1];
        }  
        // rep(j,10) dbg(dp[j].val());
    }
    mint ans = 0;
    rep(j,m+1) ans += dp[j];
    cout << ans.val() << endl;
    return 0;

}
0