리셋 되지 말자

[백준 11050] 이항 계수 1 - 수학, 구현, 조합론 본문

알고리즘

[백준 11050] 이항 계수 1 - 수학, 구현, 조합론

kyeongjun-dev 2021. 12. 30. 14:19

코드1 - 반복문

def solution(n, k):
    answer = 0
    tmp = 1
    for i in range(n, n-k, -1):
        tmp = tmp * i
    answer += tmp
    tmp = 1
    for i in range(k, 0, -1):
        tmp = tmp * i
    
    print(answer//tmp)


n, k = map(int, input().split())
solution(n, k)

 

코드2

from math import factorial
def solution(n, k):
    print(factorial(n)//(factorial(n-k)*factorial(k)))



n, k = map(int, input().split())
solution(n, k)

 

설명

  • 이항계수 그대로 구현하면 된다.
  • 파이썬에서 math 패키지의 팩토리얼을 사용하면 더 쉽게 구현할 수 있다고 한다.
Comments