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

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

 

문제 풀어보기: https://school.programmers.co.kr/learn/courses/30/lessons/12977?language=javascript 

 

풀이보기

function isPrime(n) {
  if (n < 2) return false;

  for (let i = 2; i <= Math.sqrt(n); i++) {
    if (n % i === 0) {
      return false;
    }
  }
  return true;
}

function solution(nums) {
  let count = 0;

  for (let i = 0; i < nums.length - 2; i++) {
    for (let j = i + 1; j < nums.length - 1; j++) {
      for (let k = j + 1; k < nums.length; k++) {
        let sum = nums[i] + nums[j] + nums[k];
        if (isPrime(sum)) {
          count++;
        }
      }
    }
  }

  return count;
}

나는 소수 판별하는 함수를 따로 만들어줬다. 같이 쓰자니 너무 복잡해보여서 따로 빼줬다.

 

소수 판별 함수는 일단 판별할 숫자가 2보다 작으면  1 또는 음수이기 때문에 미리 false 처리해주고,

for문을 이용해서 판별할 숫자가 i와 나누어 떨어지면 소수가 아니기 때문에 false로 처리한다.

i가 2부터 시작하는 이유는 소수 판별을 위해 먼저 false 값을 처리해주기 위함이다.

 

이제 본 문제를 풀어본다면,

먼저 소수의 개수를 담을 count 변수를 선언해줬다.

 

그 다음 for문은 사실 GPT의 도움을 받았다 ㅠㅠ

딱 봤을 때는 이해가 안가지만 손으로 풀어보면 왜 그렇게 작성한건지 이해가 간다.

 

예시로, nums = [1, 2, 3, 4] 라고 한다면,

 

i = 0일 때,

j = 1, k = 2 : nums[0] + nums[1] + nums[2] = 1 + 2 + 3 = 6

j = 1, k = 3 : nums[0] + nums[1] + nums[3] = 1 + 2 + 4 = 7 (소수)

j = 2, k =3 : nums[0] + nums[2] + nums[3] = 1 + 3 + 4 = 8

 

i = 1일 때,

j = 2, k =3 : nums[1] + nums[2] + nums[3] = 2+ 3 + 4 = 9

 

이므로 소수는 7, 1개만 있다.

 

이런 문제는 공식처럼 외워두는게 좋은건지 .. 도저히 내 머리에서 바로 떠오르지는 못할 것 같다.