[프로그래머스] 소수 만들기

문제 설명

주어진 숫자 중 3개의 수를 더했을 때 소수가 되는 경우의 개수를 구하려고 합니다. 숫자들이 들어있는 배열 nums가 매개변수로 주어질 때, nums에 있는 숫자들 중 서로 다른 3개를 골라 더했을 때 소수가 되는 경우의 개수를 return 하도록 solution 함수를 완성해 주세요.

제한사항

• nums에 들어있는 숫자의 개수는 3개 이상 50개 이하입니다.
• nums의 각 원소는 1 이상 1,000 이하의 자연수이며, 중복된 숫자가 들어있지 않습니다.

입출력 예

nums	result
[1,2,3,4]	1
[1,2,7,6,4]	4

나의 풀이

function solution(nums) {
  // combinations 모든 경우의 수 배열을 구해주는 custom 재귀함수
  function combinations(arr, num) { 
    const result = [];

    if (num === 1) {
      return arr.map((v) => [v]);
    }

    arr.forEach((fixed, index, origin) => {
      const rest = origin.slice(index + 1);
      const combinationsArr = combinations(rest, num - 1);
      const attached = combinationsArr.map((v) => [fixed, ...v]);
      result.push(...attached);
    });

    return result;
  }

  let answer = 0;
  // allCombinationArr 소수 판별 custom 함수
  const allCombinationArr = combinations(nums, 3).map((nums) =>
    nums.reduce((a, c) => a + c, 0)
  );

  allCombinationArr.forEach((num) => {
    let isPrime = true;
    if (num < 2) {
      isPrime = false;
    } else {
      for (let i = 2; i <= Math.floor(Math.sqrt(num)); i++) {
        if (num % i === 0) {
          isPrime = false;
          break;
        }
      }
    }

    if (isPrime) {
      answer++;
    }
  });

  return answer;
}