[코딩테스트]프로그래머스 - 소수 만들기

소수 만들기

문제 설명

주어진 숫자 중 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

입출력 예 설명

입출력 예 #1

[1,2,4]를 이용해서 7을 만들 수 있습니다.

입출력 예 #2

[1,2,4]를 이용해서 7을 만들 수 있습니다.

해결한 나의 코드

function solution(nums){
  var count = 0
  var temp = []
  for(var i=0; i<nums.length; i++){
    for(var j=i+1; j<nums.length; j++){
      for(var k=j+1; k<nums.length; k++){
        var sum = nums[i]+nums[j]+nums[k]
        temp.push(sum)
      }
    }
  }
  for(var l=0; l<temp.length; l++){
    if(primeNumber(temp[l])){
      count++
    }
  }
  return count
}

function primeNumber(nums){
  for(var i=2; i*i<=nums; i++){
    if(nums % i === 0) return false
  }
  return true
}

나의 알고리즘

1. 주어진 nums 배열의 원소 3개를 골라 구할 수 있는 모든 합을 구한다.
2. 이를 temp 배열에 넣는다.
3. temp 원소가 소수인지 아닌지 판별하는 함수를 만든다.
4. 만약 소수라면, count++ 하여 count를 반환한다.
   ※ 소수가 없다면 0을 반환한다.

프로그래머스에서 주어준 코드의 answer 값이 -1로 되어있어서 '소수가 없으면 -1을 반환해야 하는구나'라고 생각했다. 근데 채점하니 3개가 틀려버림... 알고보니 소수가 없으면 0을 반환해야 하는 거였다.

질문하기 없었으면 그저 고민만 하면서 허무하게 시간을 보냈을수도 !