結果
問題 |
No.2381 Gift Exchange Party
|
ユーザー |
|
提出日時 | 2023-07-21 07:19:11 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 98 ms / 2,000 ms |
コード長 | 517 bytes |
コンパイル時間 | 169 ms |
コンパイル使用メモリ | 82,136 KB |
実行使用メモリ | 64,640 KB |
最終ジャッジ日時 | 2024-09-21 09:40:54 |
合計ジャッジ時間 | 2,519 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
n, p = map(int, input().split()) mod = 998244353 fact = [1] * (n + 1) inv = [1] * (n + 1) finv = [1] * (n + 1) for i in range(2, n + 1): fact[i] = fact[i - 1] * i % mod inv[i] = mod - inv[mod % i] * (mod // i) % mod finv[i] = finv[i - 1] * inv[i] % mod inv_p = pow(p, mod - 2, mod) ans = 0 max_k = n // p for k in range(max_k + 1): res = fact[n] * finv[k] % mod res *= finv[n - k * p] * pow(inv_p, k, mod) % mod res %= mod ans += res ans %= mod ans = (fact[n] - ans) % mod print(ans)