백준 11401 - 이항 계수 3 (파이썬)

Answer

def power(a, b):
    if b == 1:
        return a % P
    else:
        if b % 2 == 0:
            return power(a, b // 2) ** 2 % P
        else:
            return power(a, b // 2) ** 2 * a % P

N, K = map(int, input().split())
P = 1000000007

f = [1 for _ in range(N + 1)]

for i in range(2, N + 1):
    f[i] = f[i - 1] * i % P

A = f[N]
B = f[K] * f[N - K] % P

print((A % P) * (power(B, P - 2) % P) % P)