[프로그래머스] 구슬을 나누는 경우의 수 파이썬

xianxbabx·2023년 1월 22일
0

알고리즘

목록 보기
5/14

문제 설명

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

제한사항

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

입출력 예

ballsshareresult
323
5310

입출력 예 설명

입출력 예 #1

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

  • 입출력 예 #2

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

Hint!

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

나의 코드

import math
def solution(balls, share):
    up = math.factorial(balls)
    down = math.factorial(balls-share) * math.factorial(share)
    answer = up/down
    return answer

풀이

조합을 써야 하는데, 팩토리얼을 구현하는 방법은 여러가지인데 그 중에서 math 함수를 이용하여 factorial을 구현하였다.

0개의 댓글