結果

問題 No.2791 Beginner Contest
コンテスト
ユーザー Ameesh Sethi
提出日時 2024-06-21 21:38:43
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 52 ms / 2,000 ms
コード長 722 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 122 ms
コンパイル使用メモリ 85,028 KB
実行使用メモリ 81,216 KB
最終ジャッジ日時 2026-03-06 04:55:39
合計ジャッジ時間 1,445 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

from sys import stdin
def input(): return stdin.readline()[:-1]

MOD = 998244353

class FenwickTree:
    def __init__(self, n) -> None:
        self.n = n
        self.bit = [0] * (n)
    
    def sum(self, r):
        ret = 0
        while r >= 0:
            ret += self.bit[r]
            ret %= MOD
            r = (r & (r + 1)) - 1
        return ret
    
    def add(self, idx, delta):
        while idx < self.n:
            self.bit[idx] += delta
            self.bit[idx] %= MOD
            idx |= idx + 1

def solve():
    n, k = map(int, input().split())
    tree = FenwickTree(n + 1)
    tree.add(0, 1)
    
    for i in range(1, n + 1):
        tree.add(i, tree.sum(i - k))
    print(tree.sum(n))
    
solve()
0