結果
問題 |
No.2381 Gift Exchange Party
|
ユーザー |
|
提出日時 | 2023-07-14 23:04:54 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
|
実行時間 | - |
コード長 | 595 bytes |
コンパイル時間 | 182 ms |
コンパイル使用メモリ | 82,772 KB |
実行使用メモリ | 70,012 KB |
最終ジャッジ日時 | 2024-09-16 08:12:47 |
合計ジャッジ時間 | 2,439 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge6 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 13 RE * 9 |
ソースコード
import sys N,P = map(int,input().split()) mod = 998244353 if N == 1: print(0) exit() C = 10 ** 5 + 100 fact = [1] * C fact_inv = [1] * C for i in range(2,C): fact[i] = fact[i-1] * i % mod fact_inv[-1] = pow(fact[-1],mod - 2,mod) for i in range(C - 2,1,-1): fact_inv[i] = fact_inv[i + 1] * (i + 1) % mod ans = 1 k = 1 r = pow(P,mod - 2,mod) while k * P <= N: #ans += fact[N] * pow(fact_inv[P],k,mod) * fact[k] % mod ans += fact[N] * fact_inv[N - k * P] % mod * pow(r,k,mod) * fact_inv[k] % mod ans %= mod k += 1 ans = fact[N] - ans ans %= mod print(ans)