結果

問題 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 35 ms
53,960 KB
testcase_01 AC 33 ms
52,756 KB
testcase_02 AC 54 ms
75,696 KB
testcase_03 AC 34 ms
51,944 KB
testcase_04 AC 35 ms
52,464 KB
testcase_05 AC 36 ms
52,996 KB
testcase_06 AC 66 ms
76,484 KB
testcase_07 AC 55 ms
75,528 KB
testcase_08 AC 51 ms
76,056 KB
testcase_09 AC 67 ms
76,640 KB
testcase_10 AC 67 ms
76,236 KB
testcase_11 AC 54 ms
76,388 KB
testcase_12 AC 69 ms
76,624 KB
testcase_13 AC 54 ms
75,948 KB
testcase_14 AC 49 ms
75,832 KB
testcase_15 AC 56 ms
76,460 KB
testcase_16 AC 34 ms
52,192 KB
testcase_17 AC 34 ms
53,496 KB
testcase_18 AC 57 ms
76,128 KB
testcase_19 AC 58 ms
76,612 KB
権限があれば一括ダウンロードができます

ソースコード

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