結果
問題 | No.2381 Gift Exchange Party |
ユーザー | ニックネーム |
提出日時 | 2023-07-14 22:50:18 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 315 ms / 2,000 ms |
コード長 | 382 bytes |
コンパイル時間 | 136 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 26,240 KB |
最終ジャッジ日時 | 2024-09-16 07:55:18 |
合計ジャッジ時間 | 4,021 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge6 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
n,p = map(int,input().split()) mod = 998244353; fa = [1]*(n+1); fi = [1]*(n+1) for i in range(1,n): fa[i+1] = fa[i]*(i+1)%mod fi[n] = pow(fa[n],mod-2,mod) for i in range(n,0,-1): fi[i-1] = fi[i]*i%mod def cmb(n,r): return fa[n]*fi[n-r]%mod*fi[r]%mod if 0<=r<=n else 0 x = 1; ans = fa[n]-1 for i in range(n//p): x = x*fa[p-1]%mod*cmb(n-i*p,p)%mod; ans -= x*fi[i+1]%mod print(ans%mod)