結果

問題 No.2791 Beginner Contest
ユーザー Ameesh Sethi
提出日時 2024-06-21 21:38:43
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 69 ms / 2,000 ms
コード長 722 bytes
コンパイル時間 141 ms
コンパイル使用メモリ 82,372 KB
実行使用メモリ 76,640 KB
最終ジャッジ日時 2024-06-21 21:38:46
合計ジャッジ時間 1,835 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

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