結果

問題 No.2791 Beginner Contest
ユーザー Ameesh SethiAmeesh 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()
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0