[코딩테스트 입문] 구슬을 나누는 경우의 수 (Python)

나의 풀이

def solution(balls, share):
    n = 1
    a = 1
    
    for i in range(share):
        n *= balls-i
        a *= share-i
        
    return n/a

다른 풀이

import math


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

질문하기에 있던 수학 공식을 참고했다.

공식
전체 n개 중에서 a개를 순서없이 뽑는 경우의 수
n (n-1) (n -2) 를 a개 만큼 진행한 값(분자) 나누기
a (a - 1) (a - 2) 를 a가 1이 될때까지 진행한 값(분모)