結果
問題 | No.2381 Gift Exchange Party |
ユーザー |
|
提出日時 | 2023-07-14 23:05:30 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 102 ms / 2,000 ms |
コード長 | 599 bytes |
コンパイル時間 | 1,130 ms |
コンパイル使用メモリ | 82,328 KB |
実行使用メモリ | 73,816 KB |
最終ジャッジ日時 | 2024-09-16 08:13:26 |
合計ジャッジ時間 | 2,603 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
import sysN,P = map(int,input().split())mod = 998244353if N == 1:print(0)exit()C = 2 * 10 ** 5 + 100fact = [1] * Cfact_inv = [1] * Cfor i in range(2,C):fact[i] = fact[i-1] * i % modfact_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) % modans = 1k = 1r = pow(P,mod - 2,mod)while k * P <= N:#ans += fact[N] * pow(fact_inv[P],k,mod) * fact[k] % modans += fact[N] * fact_inv[N - k * P] % mod * pow(r,k,mod) * fact_inv[k] % modans %= modk += 1ans = fact[N] - ansans %= modprint(ans)