2022 카카오 코딩테스트 lv.2 k진수에서 소수 개수 구하기 파이썬

def convert(n, k):
    s = ''
    while n:
        s += str(n%k)
        n //= k
    return s[::-1]

def isprime(num):
    if num == 1:
        return False
    i = 2
    while i*i <= num:
        if num % i == 0: return False
        i += 1
    return True

def solution(n, k):
    answer = 0
    s = convert(n,k)
    for num in s.split('0'):
        if not num: continue
        if isprime(int(num)): answer += 1
    return answer

1.  n을 k진수 변환합니다 => s

2.  s를 '0'을 기준으로 나눈 배열의 각 값이 소수인지 검사합니다