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

hye·2023년 5월 4일

AlgorithmTest

목록 보기

92/95

📖 Exam

문제 설명

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

제한 조건

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

입출력 예

balls	share	result
3	2	3
5	3	10

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

✍ Answer

class Solution {
    public int solution(int balls, int share) {
        double ballsFac = 1;
        double shareFac = 1;
        double nmFac = 1;

        for(int i = 1; i <= balls; i++){
            ballsFac *= i;
        }
        for(int i = 1; i <= share; i++){
            shareFac *= i;
        }
        if((balls-share)!=0){
            for(int i = 1; i <= (balls-share); i++){
                nmFac *= i;
            }
        } else {
            return 1;
        }
        return (int) Math.round(ballsFac/(nmFac*shareFac));
    }
}