結果

問題 No.2625 Bouns Ai
コンテスト
ユーザー 寝癖
提出日時 2024-02-10 10:22:53
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 144 ms / 2,000 ms
コード長 1,751 bytes
コンパイル時間 4,626 ms
コンパイル使用メモリ 312,880 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-09-28 17:04:18
合計ジャッジ時間 8,244 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>

using namespace std;
using namespace atcoder;

typedef long long ll;
typedef long double ld;
typedef vector<int> vi;
typedef vector<ll> vll;
typedef vector<ld> vld;
typedef vector<vi> vvi;
typedef vector<vll> vvll;

template <class T> using Vec = vector<T>;
template <class T> using Graph = Vec<Vec<T>>;

#define _overload3(_1,_2,_3,name,...) name
#define _rep(i,n) repi(i,0,n)
#define repi(i,a,b) for(int i=int(a);i<int(b);++i)
#define rep(...) _overload3(__VA_ARGS__,repi,_rep,)(__VA_ARGS__)
#define _rrep(i,n) rrepi(i,n-1,-1)
#define rrepi(i,a,b) for(int i=int(a);i>int(b);--i)
#define rrep(...) _overload3(__VA_ARGS__,rrepi,_rrep)(__VA_ARGS__)

#define all(x) (x).begin(),(x).end()
#define SORT(x) sort(all(x))
#define REVERSE(x) reverse(all(x))

template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; }

using mint = modint998244353;

mint op(mint a, mint b) { return a + b; }
mint e() { return 0; }

int main() {
    int N;
    cin >> N;

    int M = 1e5;

    vi A(N);
    rep(i, N) cin >> A[i];

    // segtree<mint, op, e> dp_prev(M+1);
    // dp_prev.set(0, 1);

    Vec<mint> dp_prev(M+1);
    dp_prev[0] = 1;

    rep(i, N) {
        Vec<mint> dp_next(M+1);
        Vec<mint> sum(M+2);
        rep(j, M+1) sum[j+1] = sum[j] + dp_prev[j];
        rep(j, A[i], M+1) {
            int l = (i>0) ? A[i-1] : 0;
            int r = (i>0) ? min(j, j+A[i-1]-A[i]) : j;
            if (!(l <= r)) continue;
            dp_next[j] = sum[r+1] - sum[l];
        }
        dp_prev = dp_next;
    }

    mint ans = 0;
    rep(i, M+1) ans += dp_prev[i];
    cout << ans.val() << endl;
}
0