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

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

프로그래머스

---

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

def solution(n, k):
    answer = 0
    
    # 진수 변환
    k_ = ''    
    while n != 0:
        n, m = divmod(n, k)
        k_ += str(m)
    
    k = k_[::-1]
    s = k.split('0')
    
    for s_ in s:
        if s_ == '' or s_ == '1':
            pass
        elif is_prime(int(s_)):
            answer += 1        
    
    return answer