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

문제) 머쓱이는 구슬을 친구들에게 나누어주려고 합니다. 구슬은 모두 다르게 생겼습니다. 머쓱이가 갖고 있는 구슬의 개수 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개를 뽑는 경우의 수 공식은 다음과 같습니다. 

 

---

풀이) 처음에 범위 생각 안하다가..된통 틀린 후 BigInteger로 변경

1. Hint에 주어진 공식의 분모와 분자가 될 BigInteger 1로 선언

2. 분모에서 (balls-share) 부분은 어차피 분자에 포함돼서 약분 가능한 값이기 때문에  n부터 n-m만큼 반복문으로 연산(분모)

3. share!의 값인 분자 연산

4. 나눠준 후 리턴

 

import java.math.BigInteger;
class Solution {
    public BigInteger solution(int balls, int share) {
        
		BigInteger top = new BigInteger("1");
		BigInteger bottom = new BigInteger("1");
		
		for(int i=balls; i>balls-share; i--) {
			top = top.multiply(BigInteger.valueOf(i));
		}
		
		for(int i=share; i>=1; i--) {
			bottom = bottom.multiply(BigInteger.valueOf(i));
		}
        return top.divide(bottom);
    }
}

 

다른 분들 풀이보니까 재귀함수로 간단하게 구현하시던데

현재로선 백날 봐도 내 지식이 될 것 같진 않아서 좀 더 공부하고 파고들어보기로 했다. ㅠ.ㅠ 하팅