[알고리즘]백준 2581번: 소수 W_node.js

문제

https://www.acmicpc.net/problem/2581

2581번: 소수

풀이

const readFileSyncAddress = '/dev/stdin';

const fs = require('fs');
let [m, n] = fs
  .readFileSync(readFileSyncAddress)
  .toString()
  .trim()
  .replaceAll(/\r/g, '')
  .split(/\n/g)
  .map((v) => +v);

const isPrime = function (x) {
  // x가 1이면 소수가 아니므로 false
  if (x === 1) return false;
  // 2부터 시작해 제곱근까지 나누어 떨어지는 수가 있으면 소수가 아니므로 false
  for (let i = 2; i <= Math.sqrt(x); i++) {
    if (x % i === 0) return false;
  }
  // 조건을 모두 통과하면 소수이므로 true
  return true;
};

const solution = function (m, n) {
  let min = Number.MAX_SAFE_INTEGER;
  let sum = 0;
  for (let i = m; i <= n; i++) {
    if (i === 2 || (i % 2 !== 0 && isPrime(i))) {
      min = Math.min(i, min);
      sum += i;
    }
  }
  let answer = sum > 0 ? `${sum}\n${min}` : -1;
  return answer;
};

console.log(solution(m, n));

isPrime 함수를 실행하기 전에, 2를 제외한 모든 소수는 홀수이므로 이 조건에 부합할 때만 isPrime 함수를 실행했다.

처음 제출했을때 i === 2 라는 조건을 추가하지 않아서 오답판정을 받았다.

i === 2라는 조건을 추가하기 위해서 if...else 문을 계속 수정했는데, 그러면 min을 비교하고 sum 을 누적 합산하는 과정을 반복하는 것이 걸렸다.

조금 생각해보니 if문의 조건을 i === 2 이거나, i가 홀수 및 isPrime이 true 일 때만 실행하는 것으로 수정하여 실행하면 되는 것이었다. 조건문을 잘 쓰면 코드를 더 읽기 쉽고 짧게 짤 수 있는 것 같다.

추가로, !== 를 자꾸 ==! 로 쓰는 버릇이 있는데, 고쳐야한다.