結果
問題 | No.823 Many Shifts Easy |
ユーザー | roaris |
提出日時 | 2021-05-31 04:59:57 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 81 ms / 2,000 ms |
コード長 | 596 bytes |
コンパイル時間 | 357 ms |
コンパイル使用メモリ | 87,104 KB |
実行使用メモリ | 78,492 KB |
最終ジャッジ日時 | 2023-08-08 15:22:12 |
合計ジャッジ時間 | 2,642 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge12 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 79 ms
78,492 KB |
testcase_01 | AC | 80 ms
78,420 KB |
testcase_02 | AC | 80 ms
78,484 KB |
testcase_03 | AC | 80 ms
78,224 KB |
testcase_04 | AC | 79 ms
78,268 KB |
testcase_05 | AC | 80 ms
78,228 KB |
testcase_06 | AC | 81 ms
78,228 KB |
testcase_07 | AC | 80 ms
78,456 KB |
testcase_08 | AC | 79 ms
78,420 KB |
testcase_09 | AC | 80 ms
78,204 KB |
ソースコード
MAX = 10**5+100 MOD = 10**9+7 fact = [0]*MAX #fact[i]: i! inv = [0]*MAX #inv[i]: iの逆元 finv = [0]*MAX #finv[i]: i!の逆元 fact[0] = 1 fact[1] = 1 finv[0] = 1 finv[1] = 1 inv[1] = 1 for i in range(2, MAX): fact[i] = fact[i-1]*i%MOD inv[i] = MOD-inv[MOD%i]*(MOD//i)%MOD finv[i] = finv[i-1]*inv[i]%MOD def C(n, r): if n<r: return 0 if n<0 or r<0: return 0 return fact[n]*(finv[r]*finv[n-r]%MOD)%MOD N, 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)//2 ans += C(N-1, K)*fact[K]*N print(ans%MOD)