結果

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

ソースコード

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
ans += N * (N-1) // 2 * comb(K,K-2) * perm(N-2,K-2)
ans %= MOD
print(ans)
0