알고리즘
[백준 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 패키지의 팩토리얼을 사용하면 더 쉽게 구현할 수 있다고 한다.