結果
| 問題 | 
                            No.2791 Beginner Contest
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 2024-06-21 22:28:18 | 
| 言語 | C++23  (gcc 13.3.0 + boost 1.87.0)  | 
                    
| 結果 | 
                             
                                AC
                                 
                             
                            
                         | 
                    
| 実行時間 | 5 ms / 2,000 ms | 
| コード長 | 476 bytes | 
| コンパイル時間 | 3,756 ms | 
| コンパイル使用メモリ | 277,564 KB | 
| 実行使用メモリ | 6,944 KB | 
| 最終ジャッジ日時 | 2024-06-21 22:28:24 | 
| 合計ジャッジ時間 | 4,817 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge5 / judge2 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 3 | 
| other | AC * 17 | 
ソースコード
#include<bits/stdc++.h>
#include<atcoder/modint>
#include<atcoder/fenwicktree>
using namespace std;
using namespace atcoder;
using mint = modint998244353;
int main() {
  int N, K;
  cin >> N >> K;
  fenwick_tree<mint> DP(N + 2);
  DP.add(0, 1);
  for (int i = K; i <= N; i++) {
    //dp[i] = dp[0] + dp[1] + ... + dp[i - K];
    if (i - K < 0) continue;
    mint Add = DP.sum(0, i - K + 1);
    DP.add(i, Add);
  }
  mint ans = DP.sum(0, N + 1);
  cout << ans.val() << endl;
}