結果
問題 |
No.2791 Beginner Contest
|
ユーザー |
|
提出日時 | 2024-06-21 21:36:35 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 669 bytes |
コンパイル時間 | 174 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 14,720 KB |
最終ジャッジ日時 | 2024-06-21 21:36:42 |
合計ジャッジ時間 | 7,008 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 WA * 2 |
other | WA * 17 |
ソースコード
class BIT: def __init__(self, n): self.n = n self.data = [0] * (n + 1) def sum(self, i): s = 0 while i > 0: s = (s + self.data[i]) % MOD i -= i & -i return s def add(self, i, x): while i <= self.n: self.data[i] = (self.data[i] + x) % MOD i += i & -i MOD = 998244353 def cnt(N, K): bit = BIT(N + 1) bit.add(1, 1) for i in range(1, N + 1): val = (bit.sum(i) - bit.sum(max(0, i - K + 1)) + MOD) % MOD bit.add(i + 1, val) return bit.sum(N + 1) N, K = map(int, input().split()) result = cnt(N, K) print(result)