[프로그래머스] 구슬을 나누는 경우의 수 (C++)

구슬을 나누는 경우의 수 / Lv.0

---

문제  설명 )

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

 

 

 

제한 사항 )

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

 

 

 

입출력 예 )

 

 

입출력 예 설명 )

입출력 예 #1

• 서로 다른 구슬 3개 중 2개를 고르는 경우의 수는 3입니다.

입출력 예 #2

• 서로 다른 구슬 5개 중 3개를 고르는 경우의 수는 10입니다.

Hint

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

 

 

풀이)

 

#include <string>
#include <vector>
 
using namespace std;
 
__int128 factorial(int n)
{
    __int128 temp = 1;
    for(int i = n; i > 0; i--)
    {
        temp *= i;
    }
    
    return temp;
}
 
__int128 factorial(int n, int a)
{
    __int128 temp = 1;
    for(int i = n; i > a; i--)
    {
        temp *= i;
    }
    return temp;
}
 
// 2^127까지 사용 가능한 __int128을 사용
int solution(int balls, int share) {
    int answer = 0;
    
    answer = factorial(balls, share) / factorial(balls - share);
    
    return answer;
}

---

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

프로그래머스