結果
問題 |
No.2381 Gift Exchange Party
|
ユーザー |
![]() |
提出日時 | 2023-07-16 15:05:36 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 63 ms / 2,000 ms |
コード長 | 1,068 bytes |
コンパイル時間 | 159 ms |
コンパイル使用メモリ | 82,424 KB |
実行使用メモリ | 74,520 KB |
最終ジャッジ日時 | 2024-09-17 12:43:47 |
合計ジャッジ時間 | 2,180 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
N, P = map(int, input().split()) MOD = 998244353 fact = [1] * (N + 1) rfact = [1] * (N + 1) r = 1 for i in range(1, N + 1): fact[i] = r = r * i % MOD rfact[N] = r = pow(fact[N], MOD - 2, MOD) for i in range(N, 0, -1): rfact[i - 1] = r = r * i % MOD def comb(n, k): return fact[n] * rfact[k] * rfact[n - k] % MOD if N < P: print(fact[N] - 1) exit() ''' 全員が最初に持っていたプレゼントをP回の操作後に持っているパターンの通り数を数え、N!から差し引く。 1. 全員がA[i] == i (i = 0..N-1) 2. P人のグループでP回の操作をして一巡するパターン、 N//Pのグループが出来うるのでそれを1~N//Pでカウントする 選択されていない人はA[i] == iとなり一通り P人のグループ内の順番通り数は (P-1)! 解説解法の2番目 ''' ans = 0 tmp = 0 tmp1 = 1 npp = N // P for i in range(N // P): tmp1 *= comb(N - i * P, P) * fact[P-1] tmp1 %= MOD tmp += tmp1 * rfact[i+1] tmp %= MOD ans = fact[N] - tmp - 1 ans %= MOD print(ans)