結果
問題 | No.823 Many Shifts Easy |
ユーザー |
![]() |
提出日時 | 2021-05-31 04:59:57 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 53 ms / 2,000 ms |
コード長 | 596 bytes |
コンパイル時間 | 339 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 61,056 KB |
最終ジャッジ日時 | 2024-11-08 21:06:45 |
合計ジャッジ時間 | 1,462 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 10 |
ソースコード
MAX = 10**5+100MOD = 10**9+7fact = [0]*MAX #fact[i]: i!inv = [0]*MAX #inv[i]: iの逆元finv = [0]*MAX #finv[i]: i!の逆元fact[0] = 1fact[1] = 1finv[0] = 1finv[1] = 1inv[1] = 1for i in range(2, MAX):fact[i] = fact[i-1]*i%MODinv[i] = MOD-inv[MOD%i]*(MOD//i)%MODfinv[i] = finv[i-1]*inv[i]%MODdef C(n, r):if n<r:return 0if n<0 or r<0:return 0return fact[n]*(finv[r]*finv[n-r]%MOD)%MODN, K = map(int, input().split())ans = C(N-1, K)*fact[K]+C(K, 2)*C(N-2, K-2)*fact[K-2]ans *= N*(N-1)//2ans += C(N-1, K)*fact[K]*Nprint(ans%MOD)