[프로그래머스] '구슬을 나누는 경우의 수' - Java
<문제 제시>
<문제설명>
머쓱이는 구슬을 친구들에게 나누어주려고 합니다. 구슬은 모두 다르게 생겼습니다.
머쓱이가 갖고 있는 구슬의 개수 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