16-wkdghdwns199

[wkdghdwns199]

🔗 문제 링크

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

✔️ 소요된 시간

30분 - factorial 함수라는 게 있네..?

✨ 수도 코드

정말 쉬운 문제인데, 다른 풀이도 있어서 풀어봤습니다!

우선 이항 계수란 이항식을 이항 정리로 전개했을 때 각 항의 계수이며, 주어진 크기의 (순서 없는) 조합의 가짓수 이다.
그래서 조합을 계산하면 되는 문제인데, 식은
nCk = n! // (k! * (n-k)!) 입니다.

그러므로 팩토리얼을 계산하는 함수를 만들어서 팩토리얼 계산 후, 조합 식으로 계산을 해서 출력을 하면 된다.

from sys import stdin

def factorial(n):
    if n == 0:
        return 1
    return n * factorial(n-1)

n, k = map(int, stdin.readline().split())
print(factorial(n) // (factorial(k) * factorial(n - k)))

파이썬 factorial math 함수가 있네요. 더 간단하게 작성해봤습니다.

import sys, math
input = sys.stdin.readline
N,K = map(int, input().split())
print(math.factorial(N) // (math.factorial(K) * math.factorial(N-K)))

📚 새롭게 알게된 내용

[alstjr7437]

예전에 푼 문제인데
저는 두가지 방법으로 풀었었네요!!

def fibo(n, k):
    if k == 0 or n == k:
        return 1
    return fibo(n-1, k) + fibo(n-1, k-1)

print(fibo(n,k))

def fibo2(n):
    if n == 0 :
        return 1
    return n * fibo2(n-1)

print(fibo2(n)// (fibo2(n-k) * fibo2(k)))

[fnzksxl]

아 재귀 국룰 문제군요 !
이 문제 처음 풀 때 nCk 어떻게 바꾸는지 기억 안 나서 공식 검색해본 기억이 나네요.. 그런데 저도 파이썬에서 math 라이브러리를 꺼내본 적이 없어서 이게 구현이 되어있는지는 처음 알았네요

import sys

input = sys.stdin.readline

def func(n):
    if n == 0:
        return 1
    return n * func(n-1)

N, K = map(int, input().split())
print(func(N) // (func(K) * factorial(N-K)))