[프로그래머스] 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개입니다. 이와 같이, 중복되는 소수를 발견하더라도 모두 따로 세어야 합니다.

제출 답안

1. 먼저 작성해야 하는것은 10진수를 n진수로 변환하는 코드이다.
   1. convert_number는 string에 있는 함수를 활용하여 숫자와 소문자를 결합한 것이다.
   2. k를 기준으로 하나의 완성된 변환된 값을 만든다.
2. 그 다음은 소수 판정 알고리즘이다.
   1. 1인경우와 해당 숫자가 배수로 나누어지는 경우 소수가 아니기에 False를 반환한다.
   2. 1에 해당하지 않으면 True를 반환한다.
3. 처음에는 문제에서 제시한 4가지의 경우를 정규식을 사용하여 match되는 것만 뽑아내려고 하였다.
   1. 하지만 자세히보니 0을 기준으로 나눈 것처럼 보였다.
   2. 그래서 0을 -으로 변경하였다.
      1. 정규식으로 숫자인 것들만 모두 뽑아내려고 했기 때문이다.
      2. 1을 같이 변환하지 않은 이유는 11같은 경우 남아야 하기 때문이다.
      3. 0으로 split할 수도 있지만 그러면 리스트안에 또 ""와같이 공백이 들어가기에 그냥 findall을 사용하였다.
      4. n_list에서 소수가 있으면 answer에 1을 더하고 return한다.

import string
import re
convert_number = string.digits+string.ascii_lowercase

def convert(num, base) :
    q, r = divmod(num, base)
    if q == 0 :
        return convert_number[r] 
    else :
        return convert(q, base) + convert_number[r]

def is_prime_number(x):
    if x == 1:
        return False
    for i in range(2, int(x ** 0.5) + 1):
        if x % i == 0:
            return False
    return True

def solution(n, k):
    n = convert(n, k)
    n = n.replace("0", "-")
    n_list = re.findall('\d+', n)
    answer = 0
    for num in n_list:
        if is_prime_number(int(num)):
            answer += 1
    return answer