結果

問題 No.823 Many Shifts Easy
コンテスト
ユーザー ntuda
提出日時 2025-10-28 21:22:31
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 42 ms / 2,000 ms
コード長 577 bytes
コンパイル時間 296 ms
コンパイル使用メモリ 82,836 KB
実行使用メモリ 60,532 KB
最終ジャッジ日時 2025-10-28 21:22:33
合計ジャッジ時間 1,591 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,K = map(int,input().split())
if N == 1:
    print(0)
    exit()

MOD = 10 ** 9 + 7
fact = [1]*(N+1)
rfact = [1]*(N+1)
r = 1
for i in range(1, N+1):
    fact[i] = r = r * i % MOD
rfact[N] = r = pow(fact[N], -1, MOD)
for i in range(N, 0, -1):
    rfact[i-1] = r = r * i % MOD

# nCk (mod MOD) を求める
def comb(n, k):
    return fact[n] * rfact[k] * rfact[n-k] % MOD

def perm(n, k):
    return fact[n] * rfact[n-k] % MOD

ans = 0
if K < N:
    ans = (N*(N+1)//2 * perm(N-1,K)) % MOD
if K > 1:
    ans += N * (N-1) // 2 * comb(K,K-2) * perm(N-2,K-2)
ans %= MOD
print(ans)
0