結果
問題 | No.2791 Beginner Contest |
ユーザー |
|
提出日時 | 2024-06-21 22:14:56 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 49 ms / 2,000 ms |
コード長 | 736 bytes |
コンパイル時間 | 407 ms |
コンパイル使用メモリ | 82,580 KB |
実行使用メモリ | 60,128 KB |
最終ジャッジ日時 | 2024-06-21 22:14:59 |
合計ジャッジ時間 | 2,623 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
import sys input = lambda :sys.stdin.readline()[:-1] ni = lambda :int(input()) na = lambda :list(map(int,input().split())) yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES") no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO") ####################################################################### n, k = na() mod = 998244353 def find_coefficient(N, k, MOD=998244353): # Initialize the coefficients array a = [0] * (N + 1) a[0] = 1 # a_0 = 1 # Compute the coefficients using the recurrence relation for n in range(1, N + 1): a[n] = a[n - 1] if n >= k: a[n] = (a[n] + a[n - k]) % MOD return a[N] print(find_coefficient(n, k))