[프로그래머스] 소수 찾기 (level 2, Python)

https://school.programmers.co.kr/learn/courses/30/lessons/42839

프로그래머스

 

아이디어

가능한 모든 조합을 구하고, 소수인지 판별하도록 구현하였다!

 

전체 코드

def solution(numbers):
    
    def helper(prefix, lefts):
        yield prefix
        if not lefts: return
        for idx, l in enumerate(lefts):
            yield from helper(prefix + [l], lefts[:idx]+lefts[idx+1:])
    
    numbers = [ x for x in numbers ]
    numbers = [ int("".join(x)) for x in list(helper([], numbers)) if x ]
    numbers = set(numbers)
    
    def isPrime(num):
        if num == 0 or num == 1: return False
        for n in range(2, num):
            if num % n == 0: return False
        return True
    
    answer = 0
    for num in numbers:
        if isPrime(num): answer += 1
    return answer