[level 0] 구슬을 나누는 경우의 수

문제 설명

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

---

제한사항

• 1 ≤ balls ≤ 30
• 1 ≤ share ≤ 30
• 구슬을 고르는 순서는 고려하지 않습니다.
• share ≤ balls

---

내 풀이

def solution(balls, share):
    ans = 1
    a = 1
    
    for i in range(1, share+1):
        a *= i
    
    while(share):
        ans *= balls
        balls -= 1
        share -= 1

    return ans/a

냅다 구현해버리기

 

다른 풀이

import math

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

라이브러리만 알면 짧게 끝낼 수 있음.. 기억이 안났을 뿐..!

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

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

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