[프로그래머스] k진수에서 소수 구하기 - JS

2022 KAKAO BLIND RECRUITMENT - k진수에서 소수 구하기

https://school.programmers.co.kr/learn/courses/30/lessons/92335

 

문제 설명
양의 정수 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

입출력 예 #1
문제 예시와 같습니다.

입출력 예 #2
110011을 10진수로 바꾸면 110011입니다. 여기서 찾을 수 있는 조건에 맞는 소수는 11, 11 2개입니다. 이와 같이, 중복되는 소수를 발견하더라도 모두 따로 세어야 합니다.

 

 

내 풀이

function solution(n, k) {
    let count = 0;

    const converted = cvtK(n, k).split("0");

    converted.forEach((num) => {
        if (isPrime(Number(num))) {
            count++;
        }
    });

    return count;
}

const cvtK = (n, k) => {
    let result = "";
    while (n > 0) {
        let cvt = n % k;
        result = cvt + result;
        n = Math.floor(n / k);
    }
    return result || "0";
}

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

//n진수 변환 toString x
//소수 판별

 

일단 가장 큰 풀이 계획으로 n진수 변환과 소수 판별을 생각했는데,

코치님께서 toString을 사용하지 않고 n진수 변환을 직접 구현해보라고 제안해주셔서 toString을 활용하지 않았다.

그냥 toString 쓰다가 직접 구현하려니 뇌정지가 왔었다 ...

 

그리고 저번 소수 판별 문제에서 제곱근 까지만 구하면 된다고 공부해서 이번엔 적용해보았다🔥

제곱근까지만 구하게되면 실행시간이 반으로 줄어든다.

 

 

코드 해설

function solution(n, k) {
    let count = 0;  // 소수의 개수를 셀 변수

    // n을 k진수로 변환한 뒤, '0'을 기준으로 나눔
    const converted = cvtK(n, k).split("0");

    // 나누어진 각 숫자들을 순회하면서 소수인지 확인
    converted.forEach((num) => {
        if (isPrime(Number(num))) {  // 각 숫자가 소수인지 확인
            count++;  // 소수라면 카운트 증가
        }
    });

    return count;  // 소수의 개수를 반환
}

// n을 k진수로 변환하는 함수
const cvtK = (n, k) => {
    let result = "";
    while (n > 0) {
        let cvt = n % k;  // n을 k로 나눈 나머지를 구해서 그 값을 result에 더함
        result = cvt + result;  // 나머지를 결과 앞에 추가
        n = Math.floor(n / k);  // n을 k로 나누어 몫을 구하고, 그 값을 n에 저장
    }
    return result || "0";  // 만약 result가 비어 있으면 "0"을 반환
}

// 숫자가 소수인지 판별하는 함수
const isPrime = (num) => {
    if (num <= 1) {
        return false;
    }
    for (let i = 2; i <= Math.sqrt(num); i++) {  // 2부터 num의 제곱근까지 반복
        if (num % i === 0) {  // num이 i로 나눠지면 소수가 아님
            return false;
        }
    }
    return true;  // 위 조건을 만족하면 소수
}

 

 

---

고수가 되는 길은 정말 멀고도 험하다🥹