結果
問題 | No.2068 Restricted Permutation |
ユーザー |
![]() |
提出日時 | 2022-09-01 14:47:37 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 87 ms / 2,000 ms |
コード長 | 537 bytes |
コンパイル時間 | 178 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 77,392 KB |
最終ジャッジ日時 | 2024-11-14 05:58:51 |
合計ジャッジ時間 | 3,315 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
mod=998244353 M=10**5*5 fac=[1]*M ninv=[1]*M finv=[1]*M for i in range(2,M): fac[i]=fac[i-1]*i%mod ninv[i]=(-(mod//i)*ninv[mod%i])%mod finv[i]=finv[i-1]*ninv[i]%mod n,k,x=map(int,input().split()) ans=(fac[n-k]-1)*ninv[2]%mod for i in range(1,k): tmp=fac[n-i+1]*(n-i)%mod*ninv[2] tmp-=fac[n-i]*(x-1) tmp+=(x-1)*fac[n-i]%mod*(i-1)%mod*fac[n-2]%mod*finv[n-1] ans+=tmp%mod*fac[n-i-1]%mod*finv[n-i] ans%=mod ans+=fac[n-k]*(x-1) ans%=mod ans-=(x-1)*fac[n-k]%mod*(k-1)%mod*fac[n-2]%mod*finv[n-1]%mod print(ans*fac[n-1]%mod)