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

생각보다 쉽게 풀었다. 구글링을 이용해서라도 답을 내려고 하는게 우선인것 같다.

 

def sosu(n):
    for i in range(2,int(n**0.5)+1):
        if n%i==0:
            return False
 
    return True    
 
 
def convert(n, base):
    T = "0123456789ABCDEF"
    q, r = divmod(n, base)
    if q == 0:
        return T[r]
    else:
        return convert(q, base) + T[r]
 
def solution(n, k):
 
 
 
    answer = 0
 
    if k!=1:
        a = convert(n,k)
    a = a.replace("0"," ")
 
    a = a.split(" ")
 
    if len(a)==0:
        return answer
 
    for i in range(0,len(a)):
        if a[i]=="" or int(a[i])==1:
            continue
        else:
            if sosu(int(a[i]))==True:
 
                answer+=1
 
    return answer