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

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

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

 

1. k진수를 구한다.

2. P는 각 자릿수에 '0' 을 포함하지 않고 양옆에 '0' 또는 아무것도 없어야 하므로 '0' 을 기준으로 쪼개어 배열을 만든다.

3. 배열 중 소수인 원소의 개수를 구한다.

 

import math

def prime(n):
    answer = True
    
    for i in range(2, int(math.sqrt(n)) + 1):
        if n % i == 0:
            answer = False
            break
    
    return answer

def solution(n, k):
    answer = 0
    P = ''
    
    while n > 0:
        P = str(n % k) + P
        n //= k
    
    array = P.split('0')
    
    for i in array:
        if i not in ['1', '']:
            if prime(int(i)):
                answer += 1
    
    return answer

print(solution(437674, 3)) # 3