[코딩테스트 Lv.0] 구슬을 나누는 경우의 수

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

---

내 풀이

[Java]

class Solution {
    public int solution(int balls, int share) {
        long answer = 1;
        int cnt = 1;
        for(int i = share + 1; i <= balls; i++){
            answer *= i;
            answer /= cnt++;
        }
        return (int) answer;
    }
}

 

[C]

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

int solution(int balls, int share) {    
    long answer = 1;
        int cnt = 1;
        for(int i = share + 1; i <= balls; i++){            
            answer *= i;
            answer /= cnt++;
        }
        return (int) answer;
    return answer;
}

 

[Python]

def combination(n):
    if n <= 1:
        return 1
    else:
        return n * combination(n - 1)

def solution(balls, share):
    answer = 1
    n = combination(balls)
    m = combination(share)
    nm = combination(balls - share)
    answer = n / (nm * m)
    return answer

 

 

---

https://school.programmers.co.kr/learn/courses/30/lessons/120840

프로그래머스