[Javascript] k진수에서 소수 개수 구하기 (92335)

문제링크

코딩테스트 연습 - k진수에서 소수 개수 구하기

 

 

접근 방법

문제의 설명대로 접근하면 된다.

1. n을 k진수로 변화한다. : n.toString(k)

2. 0P0처럼 소수 양쪽에 0이 있는 경우
    P0처럼 소수 오른쪽에만 0이 있고 왼쪽에는 아무것도 없는 경우
    0P처럼 소수 왼쪽에만 0이 있고 오른쪽에는 아무것도 없는 경우
    P처럼 소수 양쪽에 아무것도 없는 경우
    위의 조건들은 변화한 k진수를 0으로 split()하면 나뉘게 된다.

3. 소수를 판별하는 isPrime함수를 작성하여 split()된 배열을 정수로 바꾸고 소수인지 판별한다.


주의 사항

k 진수로 변환하여 0을 기준으로 split("0") 할때, 이어진 0들이 있으면 ""로 나뉘는 경우가 있다.

let n = 43765, k = 3;
const arr = n.toString(k).split("0");
// '2020000221' => ['2', '2', '', '', '', '221']​

만약 이대로 Number형으로 치환하면 ""을 0으로 인식하여 오류가 발생한다.

console.log(Number(""))
// 0​

때문에 for문에서 arr[i] === "" 일때 예외처리를 해주거나

isPrime() 함수에서 매개변수 number === 0 일때 처리를 해줘야 한다.

(후자가 더 좋을듯)

 

const solution = (n, k) => {
  let primeCnt = 0;

  const arr = n.toString(k).split("0");

  for (let i = 0; i < arr.length; i++) {
    //if(arr[i] === "") continue;
    if (isPrime(Number(arr[i]))) primeCnt++;
  }
  return primeCnt;
};

const isPrime = (number) => {
  if (number <= 1) return false;
  if (number === 2) return true;
    
  for (let i = 2; i <= Math.sqrt(number); i++) {
    if (number % i === 0) return false;
  }

  return true;
};