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

 

 

문제 링크

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

프로그래머스

 

문제

1. 양의 정수 n을 k진수로 변환하여 소수의 개수를 모두 return하라.

2. 소수의 조건은 아래와 같다.

3. 소수를 판단하는 기준은 k진법으로 보았을 때가 아닌 10진법으로 보았을 때를 만족해야 한다.
4. 1 ≤ n ≤ 1,000,000
    3 ≤ k ≤ 10

 

 

 

문제 풀이

1. Integer.toString(int, int)를 이용하여 정수 n을 k진법의 수로 변환하여 "0"을 기준으로 나눈다.

2. "100011"과 같이 "0"이 연달아 나오는 경우를 제외하기 위해 str가 ""인 경우를 체크한다. 

3. n의 최댓값인 1,000,000을 3진수로 변환할 경우 integer 범위를 넘어가기 때문에 Long 타입으로 변환하여 소수임을 체크해야 한다.

 

 

 

코드

import java.util.Arrays;
class Solution {
    public int solution(int n, int k) {
        String [] numberArr = Integer.toString(n,k).split("0");
        int cnt = (int) Arrays.stream(numberArr).filter(it->isPrime(it)).count();
        return cnt;
    }
    
    private boolean isPrime(String str){
        if("".equals(str)) return false;
        
        long n = Long.valueOf(str);
        if(n < 2) return false;
        for(long i = 2; i*i <= n; i++){
            if(n%i == 0) return false;
        }
        return true;
    }
}

 

실행 결과