[프로그래머스 Level.2] k진수에서 소수 개수 구하기 (2022 KAKAO BLIND RECRUITMENT) (Java)

문제 링크

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

프로그래머스

 

 

 

코딩테스트 연습 > 2022 KAKAO BLIND RECRUITMENT > k진수에서 소수 개수 구하기

 

 

 

문제 설명

 

 

양의 정수 n이 주어집니다. 이 숫자를 k진수로 바꿨을 때, 변환된 수 안에 아래 조건에 맞는 소수(Prime number)가 몇 개인지 알아보려 합니다.

 

• 0P0처럼 소수 양쪽에 0이 있는 경우
• P0처럼 소수 오른쪽에만 0이 있고 왼쪽에는 아무것도 없는 경우
• 0P처럼 소수 왼쪽에만 0이 있고 오른쪽에는 아무것도 없는 경우
• P처럼 소수 양쪽에 아무것도 없는 경우
• 단, P는 각 자릿수에 0을 포함하지 않는 소수입니다.
  • 예를 들어, 101은 P가 될 수 없습니다.

 

예를 들어, 437674을 3진수로 바꾸면 211020101011입니다. 여기서 찾을 수 있는 조건에 맞는 소수는 왼쪽부터 순서대로 211, 2, 11이 있으며, 총 3개입니다. (211, 2, 11을 k진법으로 보았을 때가 아닌, 10진법으로 보았을 때 소수여야 한다는 점에 주의합니다.) 211은 P0 형태에서 찾을 수 있으며, 2는 0P0에서, 11은 0P에서 찾을 수 있습니다.

 

정수 n과 k가 매개변수로 주어집니다. n을 k진수로 바꿨을 때, 변환된 수 안에서 찾을 수 있는 위 조건에 맞는 소수의 개수를 return 하도록 solution 함수를 완성해 주세요.

 

---

제한사항

• 1 ≤ n ≤ 1,000,000
• 3 ≤ k ≤ 10

 

---

 

입출력 예

n	k	result
437674	3	3
110011	10	2

 

---

 

입출력 예 설명

 

 

입출력 예 #1

 

문제 예시와 같습니다.

 

 

 

입출력 예 #2

 

110011을 10진수로 바꾸면 110011입니다. 여기서 찾을 수 있는 조건에 맞는 소수는 11, 11 2개입니다. 이와 같이, 중복되는 소수를 발견하더라도 모두 따로 세어야 합니다.

 

 

 

 

나의 코드

import java.util.*;

class Solution {
    public int solution(int n, int k) {
        int answer = 0;
        
        String str = Integer.toString(n, k);
        
        ArrayList<String> list = new ArrayList<>();
        
        StringBuilder sb = new StringBuilder();
        
        for(int i=0; i<str.length(); i++) {
            if(str.charAt(i) != '0') {
                sb.append(str.charAt(i));
            } else {
                if(sb.length() > 0) {
                    list.add(sb.toString());
                    sb.setLength(0);
                }
            }
        }
        
        if(sb.length() > 0) { // 마지막에 sb에 담은게 있을 경우
            list.add(sb.toString());
        }
        
        for(int i=0; i<list.size(); i++) {
            // n은 백만까지지만 Integer.toString(n, k)을 한 수의 일부분을 10진법으로 본다면 int를 벗어날 수 있다
            long num = Long.parseLong(list.get(i)); 
            
            if(num == 1) continue;
            
            boolean flag = true;
            
            for(int j=2; j<=Math.sqrt(num); j++) {
                if(num % j == 0) {
                    flag = false;
                    break;
                }
            }
            
            if(flag) answer++;
        }
        
        return answer;
    }
}

 

풀이

1. 우선 Integer.toString(n, k)를 이용하여 n을 k진법으로 바꿔 String 형으로 구한다.
2. StringBuilder를 생성하여 0이 아닌 수들이 붙어있는 경우 append 해서 0이 아닌 수가 나왔을 경우 list에 넣어준다. list에 넣은 후에는 StringBuilder를 setLength(0)를 통해 초기화 해준다.
3. 마지막 문자가 0이 아닌 경우엔 sb에 추가한 상태로 for문을 나왔으므로 확인해서 list에 담아준다.
4. list에 담았으니 list에 넣은 값이 소수인지 확인을 해서 소수의 개수를 구해야 한다.
5. list의 size만큼 for문을 돌면서 list에서 값을 꺼내와 long형으로 변환해준다.
   n은 100만까지 이므로 int형으로도 된다고 착각할 수도 있지만 n을 k진수로 바꾼 형태에서 특정부분을 그대로 10진법으로 본다면 int형을 벗어날 수 있다. 
6. long형으로 변환한 뒤 값이 1이라면 소수가 아니기 때문에 넘어가고 아니라면 for문을 통해 2~루트 해당값까지 나누어지는 수가 없는지 확인하여 구할 수 있다.
7. 소수인지 아닌지 판별 할 flag를 하나 만들어서 나누어지는 수가 있다면 소수가 아니기 때문에 flag를 false로 바꾸고 for문을 탈출하였다.
8. for문을 전부 돌아도 나누어지는 수가 없다면 flag가 그대로 true이므로 소수로 판단하고 answer를 증가시킨다.