Algorithms/프로그래머스

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

LEFT 2023. 4. 29. 13:28

<문제 제시>

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

<예시 입출력>


<문제 해결 과정>

경우의 수를 구하는 문제로, 팩토리얼을 구하는 문제이다.

Try 1)

// 분자의 역할 n! 구하기
int n_fac = 1;
for(int i = balls; i >= 1 ; i--){
    n_fac *= i;
}

// (n-m)! 구하기
int nm_fac = 1;
for(int i = (balls - share); i >= 1 ; i--){
    nm_fac *= i;
}

// 분모 중 m! 구하기
int m_fac = 1;
for(int i = share; i >= 1 ; i--){
    m_fac *= i;
}

answer = n_fac / (nm_fac * m_fac);

모든 테스트 통과 실패 = 정확성 42.9
각각 팩토리얼 별로 구한 후 계산을 해보았지만, 모든 테스트에 통과하지 못했다.

팩토리얼을 분모, 분자별로 각각 구해 n_fac / (nm_fac * m_fac) 처럼 식을 구하고자 하였다.

Solution 1)

public static int combination(int n, int r) {
    if (r == 0 || n == r) {
        return 1;
    } else {
        return combination(n-1, r-1) + combination(n-1, r);
    }
}

solution 함수를 제외한 팩토리얼을 구하는 함수또한 만들어야했다.

매개변수로 n과 r을 받는데, 이는 balls와 share을 뜻한다.

만약 나누는 구슬의 개수가 0개이거나 구슬의 개수와 나눠야할 구슬의 개수가 같으면 그대로 1을 반환한다.

그 외로는 재귀함수 방식으로 계속 함수를 호출하여 값을 구한다.


<전체코드>

public class balls_factorial {
	
	public static int combination(int n, int r) {
        if (r == 0 || n == r) {
            return 1;
        } else {
            return combination(n-1, r-1) + combination(n-1, r);
        }
    }

    public static int solution(int balls, int share) {
        return combination(balls, share);
    }

	public static void main(String[] args) {
		
		int balls = 5;
		int share = 3;
		
		System.out.println(solution(balls, share));
		
	}

}

재귀함수를 구현하는 문제와 팩토리얼을 구하는 문제를 한번에 배울 수 있던 알고리즘 문제였다.

재귀함수에서도 return combination(n-1, r-1) + combination(n-1, r);

처럼 매개변수의 값을 변화시켜가며 계속 호출하는 방식이 인상적이었다.


문제링크)
https://school.programmers.co.kr/learn/courses/30/lessons/120840

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr