Lv.0 구슬을 나누는 경우의 수

문제 설명

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

입출력

코드 풀이

from math import factorial as fact

def solution(balls, share):
    n = fact(balls)
    m = fact(share)
    nm = fact(balls-share)
    
    answer = n/(nm*m)
    
    return answer

• 이항 계수 공식 : n! / (m! * (n-m)!)

출처: 프로그래머스 코딩테스트 연습