[Algorithm] 백준 11050번 이항 계수 1 (Python)

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

11050번: 이항 계수 1

문제

자연수 N과 정수 K가 주어졌을 때 이항 계수 (N K)를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N과 K가 주어진다. (1 ≤ N ≤ 10, 0 ≤ K ≤ N)

출력

(N K)를 출력한다.

예제 입력 1

5 2

예제 출력 1

10

풀이

설명

• 이항 계수를 계산할 때, 분자에는 N에서 1씩 빼서 K개만큼 곱하면 된다.
• 분모는 K부터 1까지 하나씩 곱하면 된다.

코드

N, K = map(int, input().split())
numerator = 1  # 분자
denominator = 1  # 분모

for i in range(K):
    numerator *= N-i
    denominator *= K-i

result = numerator//denominator

print(result)