[프로그래머스] Lv.2 k진수에서 소수 개수 구하기 [Python, 파이썬] KAKAO

• 문제 : https://school.programmers.co.kr/learn/courses/30/lessons/92335
• 0을 기준으로 split 해서 나온 숫자들만 소수인지 판별

 

def solution(n, k):
    answer = 0
    b = ""

    while n != 0:   # 진수 변환
        b += str(n % k)
        n //= k
    
    b = b[::-1] # 문자열 뒤집기
    
    b = b.split("0")    # 0을 기준으로 자르기
    
    for i in b: # 소수 구하기
        sw = 0
        if i == "1" or i == "":
            continue
        i = int(i)
        for j in range(2, int(i ** 0.5) + 1):
            if i % j == 0:
                sw = 1
                break
        if sw == 0:
            answer += 1
        
    return answer