[프로그래머스] k진수에서 소수 개수 구하기 / Javascript

문제 설명

양의 정수 n이 주어집니다. 이 숫자를 k진수로 바꿨을 때, 변환된 수 안에 아래 조건에 맞는 소수(Prime number)가 몇 개인지 알아보려 합니다.

• 0P0처럼 소수 양쪽에 0이 있는 경우
• P0처럼 소수 오른쪽에만 0이 있고 왼쪽에는 아무것도 없는 경우
• 0P처럼 소수 왼쪽에만 0이 있고 오른쪽에는 아무것도 없는 경우
• P처럼 소수 양쪽에 아무것도 없는 경우
• 단, P는 각 자릿수에 0을 포함하지 않는 소수입니다.
  - 예를 들어, 101은 P가 될 수 없습니다.
  

예를 들어, 437674을 3진수로 바꾸면 211020101011입니다. 여기서 찾을 수 있는 조건에 맞는 소수는 왼쪽부터 순서대로 211, 2, 11이 있으며, 총 3개입니다. (211, 2, 11을 k진법으로 보았을 때가 아닌, 10진법으로 보았을 때 소수여야 한다는 점에 주의합니다.) 211은 P0 형태에서 찾을 수 있으며, 2는 0P0에서, 11은 0P에서 찾을 수 있습니다.

정수 n과 k가 매개변수로 주어집니다. n을 k진수로 바꿨을 때, 변환된 수 안에서 찾을 수 있는 위 조건에 맞는 소수의 개수를 return 하도록 solution 함수를 완성해 주세요.

제한사항

• 1 ≤ n ≤ 1,000,000

• 3 ≤ k ≤ 10

• 입출력 예

  n	k	result
  437674	3	3
  110011	10	2

풀이과정

처음 문제의 조건을 보고서 일단 0을 이용해야겠다는 생각이 들었다.
그리고 다시 문제의 조건을 찬찬히 뜯어보는데
0이 왼쪽에만 있거나 오른쪽에만 있거나 양쪽에 있거나 아예 없어도 된다는건
결국엔 0이 어디에 있든지 상관없다는 것 아닌가..?
그럼..? 0을 기준으로 숫자를 분리하고 분리된 숫자들의 소수 여부를 확인해주면 되는거 아닌가..?
어라라..? 하며 문제를 풀었더니 정답이었다.
굉장히 얼레벌레 푼 것 같은 기분이 든다.

// 소수 판별하는 함수
const isPrime = (n) => {
    for(let i = 2; i <= Math.sqrt(n); i++) {
        // 처음에 i를 2로 써서 계속 틀렸었다..ㅎㅎ 다들 오타 조심
        if(n % i === 0) {
            return false;
        }
    }
    return true;
}

const solution = (n, k) => {
    // n을 k진수로 변경하는 부분에서 n % k를 담을 배열
    const remainArr = [];
    // 0을 기준으로 분리된 숫자들을 담을 배열
    let numberArr;
    let answer = 0;
    // n을 k진수로 바꿈
    while(n > 0) {
        // 추후 split 사용하기 위해 나머지 값은 문자열로 넣어줌
        remainArr.unshift((n % k).toString());
        n = Math.floor(n / k);
    }
    // remainArr을 하나의 문자열로 만든 상태에서 0을 기준으로 문자열 자름
    // (0은 여러개 있을 수 있음 => {1, })
    numberArr = [...remainArr.join("").split(/0{1,}/g)];
    // 분리된 숫자들을 확인
    numberArr.forEach(x => {
        const num = parseInt(x);
        // num이 1보다 크고 소수일 때 answer 증가
        if(num > 1 && isPrime(num)) {
            answer++;
        }
    })

    return answer;
}

결과

1번 테케가 시간이 좀 오래 걸리긴 하는데 이정도는 나쁘지 않다고 생각..😉