[JAVA] 2022 KAKAO BLIND RECRUITMENT【k진수에서 소수 개수 구하기】

---

매우 간단했던 문제였다.

 

k진수로 변환하는 transform 함수를 정의하고 반환받은 값을 정규식으로 숫자를 추출했다.

 

에라토스 테네스 체로 소수를 판별하는 메서드도 정의해서 소수를 판별했고, 주의할 점은 문자열 숫자가 int 범위를 초과하여 이상한 답을 도출할 수 있으므로 long 타입으로 받아서 확인해야한다.

 

---

import java.util.*;
import java.util.regex.*;

class Solution {
    public int solution(int n, int k) {
        int answer = 0;

        String num = transform(n,k);
        Pattern pattern = Pattern.compile("[1-9]+");
        Matcher matcher = pattern.matcher(num);
        while(matcher.find()){
            String target = matcher.group();
            if(isPrime(Long.parseLong(target))){
                answer++;
            }
        }
        return answer;
    }
    
    boolean isPrime(long num){
        if(num==1){
            return false;
        }
        for(int i=2;i<=Math.sqrt(num);i++){
            if(num%i==0){
                return false;
            }
        }
        return true;
    }
    
    String transform(int num,int k){
        StringBuilder result = new StringBuilder();
        while(num!=0){
            result.append(num%k);
            num/=k;
        }
        return result.reverse().toString();
    }
}