[Python] 소수 찾기

코딩테스트 연습 - 소수 찾기

풀이

1. numbers로 만들 수 있는 모든 숫자를 만든다.

2. 순서에 따라 다른 숫자가 되기 때문에 순열을 사용한다.

3. 만들어진 모든 수가 소수인지 판별하고 카운트 한다.

 

from itertools import permutations as perm

def solution(numbers):
    def isPrime(n):
        if n in (0,1):
            return False
        else:
            for i in range(2, int(n**0.5)+1):
                if n % i == 0:
                    return False
            return True
    
    answer = []
    
    for i in range(1, len(numbers)+1):
        arr = list(perm(numbers, i))
        for j in range(len(arr)):
            num = int(''.join(map(str, arr[j])))
            if isPrime(num):
                answer.append(num)
    
    answer = list(set(answer))
    
    return len(answer)