저장을 습관화

프로그래머스 LV.0 구슬을 나누는 경우의 수 본문

코딩 테스트/프로그래머스 - 자바스크립트

프로그래머스 LV.0 구슬을 나누는 경우의 수

ctrs 2023. 9. 30. 20:17

프로그래머스 LV.0 구슬을 나누는 경우의 수

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

 

프로그래머스

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

programmers.co.kr

 

1. 문제 명

구슬을 나누는 경우의 수


2. 문제 설명

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


3. 제한 사항

- 1 ≤ balls ≤ 30

- 1 ≤ share ≤ 30

- 구슬을 고르는 순서는 고려하지 않습니다.

- share ≤ balls


4. 예시

balls share result
3 2 3
5 3 10


5. 기본 제공 코드

function solution(balls, share) {
    var answer = 0;
    return answer;
}


6. 제출한 내 답

//

 

6-2. VSC에 작성한 내용

const solution = (balls, share) => {
  const factorial = (n) => {
    let facResult = 1n;
    for (i = 2n; i <= n; i += 1n) {
      facResult *= i;
    }
    return facResult;
  };

  return factorial(balls) / (factorial(balls - share) * factorial(share));
};

// 테스트
console.log(solution(3, 2));
console.log(solution(5, 3));
console.log(solution(30, 30));


7. 특이사항

나처럼 수학적 지식이 부족한 사람들을 위해서였는지 큰 힌트가 있었다

서로 다른 n개 중 m개를 뽑는 경우의 수 공식

n! / ((n-m)! * m!)

 

틀렸던 방법

const solution = (balls, share) => {
  const factorial = (n) => {
    let facResult = 1;
    for (i = 2; i <= n; i++) {
      facResult *= i;
    }
    return facResult;
  };

  return factorial(balls) / (factorial(balls - share) * factorial(share));
};

제한사항에는 balls와 share의 수는 30이하로 설정되어 있다.

위 코드로 작성하면 30!을 만들어야하는데, 30!는 자바스크립트의 number의 범위 밖이다

따라서 factorial 함수 내 연산에 쓰이는 부분들에 BigInt 처리를 해주었다.

 

 

8. 다른 사람이 작성한 답

8-1. 가장 많이 쓰인 풀이법, 재귀함수

const 팩토리얼 = (num) => num === 0 ? 1 : num * 팩토리얼(num - 1)

function solution(balls, share) {
  return Math.round(팩토리얼(balls) / 팩토리얼(balls - share) / 팩토리얼(share))
}

팩토리얼을 재귀함수로 작성하는법 메모해두자

const factorial = (num) => num === 0 ? 1 : num * factorial(num - 1)

 

8-2. Array, reduce

function solution(balls, share) {
  return (
    Array.from({ length: share }, (_, i) => balls - i).reduce((a, b) => a * b) /
    Array.from({ length: share }, (_, i) => share - i).reduce((a, b) => a * b)
  );
}

 

8-3. Array, map

const factorial = num => 
    Array
        .from({ length: num }, (_, i) => i + 1)
        .reduce((a, c) => a *= c, 1)

const solution = (n, m) => 
    Math.round(factorial(n) / (factorial(n - m) * factorial(m)))

 

힌트가 없었다면 나는 풀지 못했을것 같다..