프로그래머스 - 구슬을 나누는 경우의 수 [Python]

난이도 - Lv.0

https://school.programmers.co.kr/learn/courses/30/lessons/120840

프로그래머스

[문제]

문제 설명

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

 

제한사항

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

 

입출력 예

balls	share	result
3	2	3
5	3	10

 

Hint

• 서로 다른 n개 중 m개를 뽑는 경우의 수 공식은 다음과 같습니다. 

  

[코드]

from math import factorial

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

 

서로 다른 n개중 m개를 뽑는 공식의 내장함수가 있었다.

import math

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

※ math.comb(n, k)

반복과 순서 없이 n개의 항목에서 k개의 항목을 선택하는 방법의 수를 반환하는 내장함수

k <= n 이면 n! / (k! + (n - k)!) 이고, k > n 이면 0 이다. (힌트 공식과 같음)

인자 중 하나라도 정수가 아니면 TypeError 발생, 인자 중 하나라도 음수면 ValueError 발생