結果

問題 No.2625 Bouns Ai
ユーザー zawakasu
提出日時 2024-02-09 23:11:33
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 113 ms / 2,000 ms
コード長 1,362 bytes
コンパイル時間 2,080 ms
コンパイル使用メモリ 198,488 KB
最終ジャッジ日時 2025-02-19 04:11:42
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>




namespace zawa {

using i16 = std::int16_t;
using i32 = std::int32_t;
using i64 = std::int64_t;
using i128 = __int128_t;

using u8 = std::uint8_t;
using u16 = std::uint16_t;
using u32 = std::uint32_t;
using u64 = std::uint64_t;

using usize = std::size_t;

} // namespace zawa


namespace zawa {

void SetFastIO() {
    std::cin.tie(nullptr)->sync_with_stdio(false);
}

void SetPrecision(u32 dig) {
    std::cout << std::fixed << std::setprecision(dig);
}

} // namespace zawa
using namespace zawa;

#include <atcoder/modint>
using mint = atcoder::modint998244353;

int main() {
    SetFastIO();

    int n; std::cin >> n;
    std::vector<int> a(n);
    for (auto& x : a) std::cin >> x;
    
    const int m{100000};
    std::vector<mint> dp(m + 1);
    for (int i{a[0]} ; i <= m ; i++) dp[i] = mint::raw(1);
    for (int i{1} ; i < n ; i++) {
        std::vector<mint> next(m + 1);
        std::vector<mint> sum(m + 2);
        for (int j{} ; j < m + 1 ; j++) {
            sum[j + 1] = sum[j] + dp[j];
        }
        for (int j{a[i]} ; j <= m ; j++) {
            int index{std::max(0, j - a[i] + a[i - 1])};
            index = std::min(index, j);
            next[j] = sum[index + 1];
        }
        dp = std::move(next);
    }
    mint ans{std::reduce(dp.begin(), dp.end(), mint{})};
    std::cout << ans.val() << '\n';
}
0