[프로그래머스] 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가 될 수 없습니다.

 

풀이

1. Integer.toString({10진수 수}, {k진법의 k}); 를 이용해 K진법으로 변환

2. StringTokenizer를 이용하여 0으로 문자열 분리

3. isPrime함수를 통해 소수 판별

 

코드

import java.util.*;
import java.io.*;

class Solution {
    public int solution(int n, int k) {
        int answer = -0;
        String str = Integer.toString(n, k); // k진법 변환
        StringTokenizer st = new StringTokenizer(str, "0");
        while(st.hasMoreTokens()){
            String cur = st.nextToken();
            if(cur.equals("")) continue;
            long num = Long.parseLong(cur);
            if(isPrime(num)) answer++;
        }
        System.out.print(str);
        return answer;
    }
    public boolean isPrime(long num){
        if(num==1) return false;
        if(num==2) return true;
        for(int i=2; i<=(int)Math.sqrt(num); i++){
            if(num%i==0) return false;
        }
        return true;
    }
}