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

문제 설명

 

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

 

 

입출력 예 설명

입출력 예 #1

• 서로 다른 구슬 3개 중 2개를 고르는 경우의 수는 3입니다.

입출력 예 #2

• 서로 다른 구슬 5개 중 3개를 고르는 경우의 수는 10입니다.

Hint

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

 

 

제출 내역

import math

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

 

math.factorial(n): x의 계승을 계산

math.comb(n, m): 조합 계산 - n개 중 m개를 선택하는 경우의 수

 

 

다른 풀이

from math import factorial as f

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

 

 

프로그래머스 코딩테스트 입문 Day9 수학, 문자열, 해시, 완전탐색, 조건문

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

프로그래머스