[프로그래머스] 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

 

 

입출력 예 #2

 

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

 

 

---

 

코드 설명

 

 

1) n을 k진수로 변환합니다. while 반복문 안에서 n이 0보다 큰 동안 n을 k로 나눈 나머지를 문자열로 변환하여 change에 저장합니다. 이때 n은 k로 나눈 몫으로 초기화시켜주어야 합니다. 반복문이 종료되면 change를 역순으로 다시 저장합니다.

 

2) change를 구성하는 글자를 하나씩 살피며 만약 change[i]가 "0"이라면 change의 idx(=0)부터 "0" 직전까지 잘라 num 배열에 저장합니다. 이후 idx는 "0"이 등장한 인덱스로 초기화해줍니다. 맨 마지막에는 change의 idx부터 끝까지 잘라 num 배열에 저장합니다.

 

3) 이후 num에 저장되어있는 수들이 소수인지 판단하여 answer을 증가시켜주면 되겠습니다. 소수인지 판단하는 반복문을 작성할 때는 시간 단축을 위해 제곱근을 활용했습니다.

 

 

def solution(n, k):
    answer = 0; idx = 0
    change = ""
    num = []
    
    while n > 0 :
        mod = n % k
        change += str(mod)
        n = n//k
    change = change[::-1]
    
    for i in range(len(change)) :
        if change[i] == "0" :
            num.append(int(change[idx:i]))
            idx = i
        if i == len(change)-1 :
            num.append(int(change[idx:]))
    
    for i in range(len(num)) :
        check = 0
        if num[i] < 3 :
            if num[i] == 2 : answer += 1
            continue
        for j in range(2,int(num[i]**(1/2))+1) :
            if num[i] % j == 0 :
                check = 1
                break
        if check == 0 :
            answer += 1
        
    return answer