코딩테스트 연습 > 코딩테스트 입문 > 구슬을 나누는 경우의 수
머쓱이는 구슬을 친구들에게 나누어주려고 합니다. 구슬은 모두 다르게 생겼습니다. 머쓱이가 갖고 있는 구슬의 개수 balls와 친구들에게 나누어 줄 구슬 개수 share이 매개변수로 주어질 때, balls개의 구슬 중 share개의 구슬을 고르는 가능한 모든 경우의 수를 return 하는 solution 함수를 완성해주세요.
| balls | share | result |
|---|---|---|
| 3 | 2 | 3 |
| 5 | 3 | 10 |
class Solution {
public double solution(int balls, int share) {
double answer = 1;
for(int i = 0; i<share; i ++){
answer *= (balls-i);
answer /= (i+1);
}
return answer;
}
}
class Solution {
public int solution(int balls, int share) {
return combination(balls, share);
}
public static int combination(int balls, int share) {
if (balls == share || share == 0) return 1;
return combination((balls - 1), (share - 1)) + combination(balls - 1, share);
}
}
반복문을 통해 분자, 분모에 각각 곱하고 나누어 계산하였다.
ex. balls: 5, share: 3 일 때
분자는 5x4x3이고, 분모는 3x2x1이다 !! 이걸 수식으로 잘 넣어서.. 코드를 구현하였다.
int형은 중간 계산에서 overflow 가능성이 있기 때문에 double형으로 계산하였다.
재귀함수를 사용하여 풀었다..
조합(Combination) 문제이므로 nCk = n! / (k! * (n-k)!) 공식을 사용
nCk = (n-1)C(k-1) + (n-1)Ck이 문제에서는 반복문이 훨씬 시간이 덜 걸리고 효율적이다.
그런데 내가 재귀 문제를 진짜 못해서 이해해보려고 오랜 시간 붙잡고 있었다.. 재귀는 답을 보면 풀리는데 직접 작성해보려고 하면 너무 어렵다ㅠㅠ
base 조건부터 n=0이나 1일때를 먼저 생각하고,
n을 하나씩 키워가면서 재귀식을 만들어가면 된다는데....
말이 쉽지 ㅠㅠ 그리고 특히 한 번 답을 보고 나면 더 하기 싫어지는 재귀,,