結果
問題 | No.2791 Beginner Contest |
ユーザー |
![]() |
提出日時 | 2024-06-21 21:45:27 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 47 ms / 2,000 ms |
コード長 | 583 bytes |
コンパイル時間 | 202 ms |
コンパイル使用メモリ | 82,500 KB |
実行使用メモリ | 63,528 KB |
最終ジャッジ日時 | 2024-06-21 21:45:29 |
合計ジャッジ時間 | 2,013 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
def comb_lut(N, MOD):F = [1]*(N+1)for i in range(N):F[i+1] = F[i] * (i+1) % MODInvF = [1]*(N+1)InvF[N] = pow(F[N], -1, MOD)for i in range(N)[::-1]:InvF[i] = InvF[i+1] * (i+1) % MODdef comb(n, r):assert 0 <= n <= N and 0 <= rif 0 <= r <= n: return F[n] * InvF[r] % MOD * InvF[n-r] % MODelse: return 0return combMOD = 998_244_353N, K = map(int, input().split())comb = comb_lut(N, MOD)ans = 0for q in range(N+1):if q*K > N:breakterm = comb(N - q*K + q, q)ans += termans %= MODprint(ans)