結果
問題 | 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=998244353M=10**5*5fac=[1]*Mninv=[1]*Mfinv=[1]*Mfor i in range(2,M):fac[i]=fac[i-1]*i%modninv[i]=(-(mod//i)*ninv[mod%i])%modfinv[i]=finv[i-1]*ninv[i]%modn,k,x=map(int,input().split())ans=(fac[n-k]-1)*ninv[2]%modfor 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%=modans+=fac[n-k]*(x-1)ans%=modans-=(x-1)*fac[n-k]%mod*(k-1)%mod*fac[n-2]%mod*finv[n-1]%modprint(ans*fac[n-1]%mod)