[프로그래머스] k진수에서 소수 개수 구하기 (JAVA 풀이)

 

문제 접근

 

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

프로그래머스

 

그렇게 어려운 난이도의 문제는 아니었음

 

=> 정말 문제에 주어진대로 구현하고 자료형을 잘 사용하면 풀 수 있음

 

문제 풀이

 

우선 진법을 바꿀 때 0이 포함되지 않는 값의 길이가 int의 최댓값을 넘는 경우가 있기 때문에 long타입 사용을 결정하고 시작했음

 

그 다음 크게 2가지의 구현으로 나누었는데

isPrime이라는 소수인지를 판별해주는 함수와 changeNumber라는 진법을 변환하는 함수임

 

사실 toString을 사용하면 쉽게 진법을 바꿀 수 있었는데 이를 몰라서 직접 함수를 구현했음

Long.toString(long n, int k)은 숫자 n을 k진법 문자열로 바꿔주는 역할을함

 

isPrime을 구현할 때는 Math.sqrt(제곱근)을 사용하여 탐색 범위를 축소했음

 

주어진 수를 k진법으로 바꾼 후 이를 split을 사용해서 0을 기준으로 String 배열을 만들었음

 

=> 이 때 정규표현식을 사용해서 split("0+")을 사용하면 1001 같이 0이 연속 2개 나오는경우도 "1", "1" 로 쪼개지지만 해당 내용을 몰랐어서 split("0")으로 나눈 후에 ""(빈 문자열)이면 continue를 하는 방식으로 진행함

 

그리고 나눈 수들을 돌면서 소수라면 정답 answer을 +1 해서 반환하며 마무리함

 

전체 코드

 

class Solution {
    
    int n, k;
    
    boolean isPrime(long num) {
        
        if(num < 2) {
            return false;
        }
        
        for(long i = 2; i <= (long) Math.sqrt(num); i++) {
            if(num % i == 0) {
                return false;
            }
        }
        
        return true;
    }
    
    String changeNumber() {
        StringBuilder sb = new StringBuilder();
        
        while(n != 0) {
            sb.append(n % k);
            n /= k;
        }
        
        
        return sb.reverse().toString();
    }
    
    public int solution(int n, int k) {
        
        
        int answer = 0;
        this.n = n;
        this.k = k;
        
        String number = changeNumber();
        
        String[] arr = number.split("0");
        
        for(String s : arr) {
            
            if(s.equals("")) continue;
            
            if(isPrime(Long.parseLong(s))) {
                answer++;
            }
        }
        
        return answer;
    }
}