[백준] 11050번 - 이항 계수 1

📩 출처

문제

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

입력

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

출력

(N, K)를 출력한다.

👉 생각

• 이항 계수를 구하는 간단한 공식이 존재한다. (N, K)일 때 N! // K! // (N-K)!
• math 모듈로 펙토리얼을 사용하거나 직접 만들어서 사용하면 된다!

import math
n, k = map(int, input().split())
print(math.factorial(n)//math.factorial(k)//math.factorial(n-k))

# 직접 만들기
def f(n):
    ans = 1
    for i in range(2, n+1):
        ans *= i
    return ans
def find(n, r):
    return f(n) // f(r) // f(n-r)