⬛ [Programmers] [Python] [2022 KAKAO BLIND RECRUITMENT] [Level2] K진수에서 소수 개수 구하기

문제

https://school.programmers.co.kr/learn/challenges?order=acceptance_desc&statuses=unsolved&levels=2%2C3%2C1&languages=python3 

코딩테스트 연습 | 프로그래머스 스쿨

풀이

# 프로그래머스 2단계 - k진수에서 소수 개수 구하기
import math
def isPrime(n):
    if n == 1:
        return False
    for i in range(2, int(math.sqrt(n)) + 1):
        if n % i == 0:
            return False
    return True

def k_digit(n, k):
    rev_base = ''
    while n > 0:
        n, mod = divmod(n,k)    
        rev_base += str(mod)
    return rev_base[::-1]

def solution(n, k):
    answer = 0
    k_digit_number = k_digit(n,k)
    n_list = k_digit_number.split('0')
    for i in n_list:
        if i != "":
            if isPrime(int(i)):
                answer += 1    
    return answer
print(solution(437674, 3))

n을 k진수로 바꾸는 함수를 k_digit 이라고 만들고 

n이 소수인지 판별하는 함수를 isPrime을 선언하였다.