結果
問題 | No.2791 Beginner Contest |
ユーザー |
![]() |
提出日時 | 2024-06-21 21:39:35 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 45 ms / 2,000 ms |
コード長 | 344 bytes |
コンパイル時間 | 148 ms |
コンパイル使用メモリ | 82,420 KB |
実行使用メモリ | 61,592 KB |
最終ジャッジ日時 | 2024-06-21 21:39:38 |
合計ジャッジ時間 | 1,791 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
N, K = map(int, input().split()) dp = [0]*(N+1) imos = [0]*(N+1) MOD = 998244353 dp[0] = 1 for i in range(N): if i > 0: imos[i] += imos[i-1] imos[i] %= MOD dp[i] = imos[i] if i+K <= N: imos[i+K] += dp[i] imos[i+K] %= MOD imos[-1] += imos[-2] imos[-1] %= MOD dp[-1] = imos[-1] print(sum(dp)%MOD)