結果
問題 |
No.2791 Beginner Contest
|
ユーザー |
|
提出日時 | 2025-02-14 10:24:01 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 44 ms / 2,000 ms |
コード長 | 395 bytes |
コンパイル時間 | 454 ms |
コンパイル使用メモリ | 82,548 KB |
実行使用メモリ | 61,492 KB |
最終ジャッジ日時 | 2025-02-14 10:24:04 |
合計ジャッジ時間 | 2,294 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
n, k = map(int, input().split()) mod = 998244353 # dp[x] = レーティング x までの推移の数 dp = [0] * (n + 1) # prefix[x] = dp[0] + dp[1] + ... + dp[x] (累積和) prefix = [0] * (n + 1) dp[0] = 1 prefix[0] = 1 for x in range(1, n + 1): if x < k: dp[x] = 1 else: dp[x] = (1 + prefix[x - k]) % mod prefix[x] = (prefix[x - 1] + dp[x]) % mod print(dp[n])