프로그래머스 level2 [k진수에서 소수 개수 구하기] 파이썬

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

코딩테스트 연습 - k진수에서 소수 개수 구하기

 

처음에는 python 내장함수 split을 사용하려고 했는데, '10011'과 같이 0이 연속으로 있는 경우에는 빈 문자열이 발생하여 오류가 발생하였다.

# 프로그래머스 k진수에서 소수 개수 구하기
def isprime(v):
    for i in range(2, int(v**0.5)+1):
        if v % i == 0:
            return False

    return True


def solution(n, k):
    answer = 0
    knum = ''
    while (n != 0):
        num = n % k
        n = n//k
        knum += str(num)
    knum = knum[::-1]
    lst = []
    curr = ''
    for i in range(len(knum)):
        if knum[i] == '0' and curr != '':
            lst.append(curr)
            curr = ''
        elif knum[i] != '0':
            curr += knum[i]
    if curr != '':
        lst.append(curr)
    for n in lst:
        n = int(n)
        if n > 1 and isprime(n):
            answer += 1
    return answer

먼저 while문을 통해 k진수를 구하고, 그 값을 knum에 저장한다. 그 후, 0을 기준으로 문자를 분리하고, 각 문자가 소수인지 isprime()함수를 이용하여 판별한다.