[프로그래머스] 2022 카카오 - k 진수에서 소수 개수 구하기 - 파이썬 풀이

나의 코드

def convert(num, base):
    import string
    tmp = string.digits + string.ascii_lowercase
    q, r = divmod(num, base)
    if q == 0:
        return tmp[r]
    else:
        return convert(q, base) + tmp[r]


def isprime(n):
    for i in range(2, int(n ** .5) + 1):
        if n % i == 0:
            return False
    return True

def solution(n, k):
    answer = 0
    cur = convert(n, k).split("0")

    for i in cur:
        if len(i) == 0 or int(i) < 2:
            continue

        if isprime(int(i)):
            answer += 1

    return answer