[백준 알고리즘] 이항계수1(11050)

https://www.acmicpc.net/problem/11050

이항계수 문제를 풀기 위해서 이항계수 수식을 먼저 알아야 합니다.

$$\binom{N}{K} = \frac{N!}{K!(N-K)!}$$

파이썬 코드로 풀어봤습니다.

def factorial(x):
    if x == 1:
        return 1
    else:
        return x * factorial(x-1)

재귀함수를 이용하여 팩토리얼을 계산합니다.

위 함수가 실행되면 x * (x-1) * (x-2) * ... * 2 * 1 을 계산하게 됩니다.

def binomial(N, K):
    return factorial(N) / (factorial(K) * factorial(N - K))

이항계수를 계산하는 함수입니다.

이전에 작성한 팩토리얼 함수를 이용하면 간단하게 이항계수 공식에 따라 만들 수 있습니다.

N, K = map(int, input().split())
print(int(binomial(N, K)))

두 숫자를 한 줄로 입력받기 때문에 split()함수로 분리합니다.

문자열로 입력받은 값을 map()함수를 이용하여 정수형으로 변환 후 위에서 작성한 이항계수 함수로 답을 출력합니다.

def factorial(x):
    if x == 1:
        return 1
    else:
        return x * factorial(x-1)

def binomial(N, K):
    return factorial(N) / (factorial(K) * factorial(N - K))

N, K = map(int, input().split())
print(int(binomial(N, K)))