Programmers Java Lv.0 (27) - 구슬을 나누는 경우의 수

구슬을 나누는 경우의 수

---

문제 설명

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

---

제한사항

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

---

입출력 예

balls	share	result
3	2	3
5	3	10

---

입출력 예 설명

입출력 예 #1

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

입출력 예 #2

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

---

Hint

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

---

Solution

처음엔 factorial 함수를 만들어서 접근했지만, 숫자가 너무 크게 나와서 일부 케이스에서는 오버플로우가 발생했습니다.

class Solution {
    public int solution(int balls, int share) {
		 double result = factorial(balls) / (factorial(balls-share) * factorial(share));
        return (int) result;
    }
    
    private static double factorial(int n){
        if (n == 0) {
            return 1;
        }
        
        double result = 1;
        for(int i = 1; i <= n; i++){
            result *= i;
        }
        return result;
    }
}

for문을 사용해 각 반복 단계에서 answer에 (balls - i) / (i+1)을 곱하는 방법으로 문제를 풀었습니다.

class Solution {
    public int solution(int balls, int share) {
         double answer = 1;

        for(int i = 0; i < share; i++){
            answer = answer * (balls - i) / (i+1);
        }
        return (int)answer;
    }
}