[python] 프로그래머스 - k진수에서 소수 개수 구하기

출처 : 프로그래머스, https://school.programmers.co.kr/learn/courses/30/lessons/92335

프로그래머스

 

 

먼저, 생각해야 할 부분

 

1. 조건에 해당하는 숫자문자열을 뽑아내기 위해서 가장 쉬운 방법은 무엇일지 생각해보자.

2. 어떤 게 더 쉬울지 생각해보자.

     조건을 만족하는 소수를 찾는게 쉬울까, 아니면 일단 조건에 맞는 숫자를 찾고 그 중 소수를 골라내는게 더 쉬울까?

 

---

 

풀이

 

# 진수 변환

def changeToK(n, k):
    k_string = ''
    
    while n != 0:
        k_string = str(n % k) + k_string
        n //= k
            
    return k_string

# 조건 만족
# 그냥 0으로 끊으면 모든 조건에 포함.

def isSatisfy(num_string):
    return [string for string in num_string.split('0') if len(string) > 0]

# 소수 판별

def isPrime(num):
    num = int(num)
    
    if num == 1:
        return False
    
    for i in range(2, int(num**0.5 + 1)):
        if num % i == 0:
            return False
    
    return True

def solution(n, k):
    return len([i for i in isSatisfy(changeToK(n, k)) if isPrime(i)])

 

말을 어렵게 해놨는데 그냥 0으로 자른 배열이 모든 조건에 포함된다.

---

 

Github

GitHub - bh2980/Algorithm-Problem: 알고리즘 풀이 흔적들