[백준] 11050 - Python

문제

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

입력

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

출력

(NK)(\binom{N}{K})를 출력한다.

풀이

이항 계수가 무엇이고 어떻게 계산하는지만 알면 쉽게 풀 수 있는 문제이다.

import sys

n, k = map(int, sys.stdin.readline().rstrip().split())

a = 1
b = 1

while k > 0:
    a = a * n
    b = b * k
    n = n - 1
    k = k - 1

print(a//b)