구슬을 나누는 경우의 수

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

프로그래머스

 

[문제 설명]

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

 

제한사항

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

[나의 풀이]

import math
def solution(balls, share):
    a = math.factorial(balls)
    b = math.factorial(share) * math.factorial(balls-share)
    return a//b

solution(5,2)

처음에 이렇게 썼는데 통과가 안됐음...

구글코랩으로 하니깐 되는데 뭐지..? 싶었는데 다시해보니깐 되네? 이 자식

 

그리고 return에서 a/b 하면 10.0이라고 나옴 a//b라고 해야 10으로 떨어짐

from math import factorial as fac
def solution(balls, share):
    a = fac(balls)
    b = fac(share) * fac(balls-share)
    return a//b

최종적으로는 이렇게 쓰긴 함

 

[다른 풀이]

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)