[11050번] 이항계수1 코딩

문제

자연수 $$N과 정수 $$K가 주어졌을 때 이항 계수 $<span\; style="font-family:\; "맑은\; 고딕",\; sans-serif;">(</span>\genfrac{}{}{0ex}{}{\mathrm{<span\; style="font-family:\; "맑은\; 고딕",\; sans-serif;">N</span>}}{\mathrm{<span\; style="font-family:\; "맑은\; 고딕",\; sans-serif;">K</span>}}<span\; style="font-family:\; "맑은\; 고딕",\; sans-serif;">)</span>$를 구하는 프로그램을 작성하시오.

---

코딩

n,k=list(map(int,input().split()))
def test(num,k):
    tmp=1
    cnt=k
    ret=1
    for _ in range(k): #N부터 분자 수많큼 곱하고(N*(N-1)*(N-2)...)
        tmp=tmp*num    #분모에 분자 팩토리얼값(ex- 3!=6)으로 나누도록 함
        ret=ret*cnt
        cnt=cnt-1
        num=num-1
    print(int(tmp/ret))
test(n,k)

수학 공부할때 쓰던 방법을 적용해서 해보니까 결과가 잘 나왔다. 다행히 ㅎㅎ

근데 왠지 더 효율적인 방법이 있을 것 같다..ㅋ