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

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

코딩테스트 연습 - k진수에서 소수 개수 구하기

[문제 풀이]

간단한 문제였는데, int 형을 사용시 런타임에러가 발생하여 더 큰수를 받을수있게 Long형을 사용하여 해결하였습니다.

 

- 입력받은 n을 String 형식으로 k진수로 변환하여 result로 반환합니다.

- count() 함수를 통해 "0"을 기준으로 split합니다.

- split된 String을 각각 Long형으로 변환하여 check()를 통해 소수인지 확인하여 answer을 갱신합니다.

 

이때, 소수를 확인하는 방법은 아래의 블로그를 참고하였습니다.

https://st-lab.tistory.com/81

JAVA [자바] - 소수 구하는 알고리즘 및 구현

 

[정답 코드]

class Solution {
    public static int answer = 0;
    public int solution(int n, int k) {
        String result = "";
        int num = n;
        int mod = 0;
        
        // k 진수로 변환
        while(num > 0) {
            mod = num%k;
            num = num/k;
            result = mod + result;
        }
        count(result);
        return answer;
    }
    public static void count(String str) {
        String[] strArr = str.split("0");
        for(int i=0;i<strArr.length;i++) {
            if(strArr[i].length() > 0) {
                Long num = Long.parseLong(strArr[i]);
                if(check(num)) {
                    answer++;
                }
            }
        }
    }
    public static boolean check(Long num) {
        if(num < 2L) {
            return false;
        } else if(num == 2L) {
            return true;
        }
        for(long i=2L;i<=Math.sqrt(num);i++) {
            if(num % i == 0) {
                return false;
            }
        }
        return true;
    }
}