結果
問題 |
No.823 Many Shifts Easy
|
ユーザー |
|
提出日時 | 2020-04-21 21:31:54 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 700 ms / 2,000 ms |
コード長 | 867 bytes |
コンパイル時間 | 140 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 12,032 KB |
最終ジャッジ日時 | 2024-10-09 10:21:14 |
合計ジャッジ時間 | 2,725 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 10 |
ソースコード
INF = 10 ** 9 MOD = 10 ** 9 + 7 import sys sys.setrecursionlimit(100000000) dy = (-1,0,1,0) dx = (0,1,0,-1) from math import factorial def main(): n,k = map(int,input().split()) ans = 0 if n == 1: print(0) return if n == 2: if k == 1: print(3) else: print(1) return if n == k: npk2 = factorial(n - 2)%MOD for i in range(1,n): ans += (npk2 * k * (k - 1)//2)*i%MOD ans %= MOD else: npk = factorial(n - 1) * pow(factorial(n - k - 1),MOD - 2,MOD) % MOD npk2 = factorial(n - 2) * pow(factorial(n - k),MOD - 2,MOD) %MOD for i in range(1,n): ans += (npk + npk2 * k * (k - 1)//2)*i%MOD ans %= MOD ans += n * npk ans %= MOD print(ans) if __name__ == '__main__': main()