[프로그래머스 lv2] k진수에서 소수 개수 구하기 - 파이썬(Python)

문제

프로그래머스

풀이

def solution(n, k):
    result = 0
    # n의 k진수 num[::-1] 구하기
    num = ''
    while n > 0:
        num += str(n % k)
        n //= k
    for p in num[::-1].split('0'):
        if p.isdigit() and int(p) > 1 : # 빈 문자열과 1은 제외
            # p가 소수인지 판별
            for i in range(2,int(int(p)**(1/2))+1):
                if int(p) % i == 0:
                    break
            else:
                result += 1
    return result

실행 결과