[코딩테스트/programmers] Lv.0 구슬을 나누는 경우의 수

문제

머쓱이는 구슬을 친구들에게 나누어주려고 합니다. 구슬은 모두 다르게 생겼습니다. 머쓱이가 갖고 있는 구슬의 개수 balls와 친구들에게 나누어 줄 구슬 개수 share이 매개변수로 주어질 때, balls개의 구슬 중 share개의 구슬을 고르는 가능한 모든 경우의 수를 return 하는 solution 함수를 완성해주세요.

 

입출력

balls	share	result
3	2	3
5	3	10

 

답

def factorial(n):
    r = 1
    for i in range(1, n + 1):
        r *= i
    
    return r

def solution(balls, share):
    return factorial(balls) // (factorial(balls - share) * factorial(share))

 

💡 math.comb( ) 

import math

def solution(balls, share):
    return math.comb(balls, share)