프로그래머스) 소수 찾기-python

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

프로그래머스

 

from itertools import permutations

def isTrue(n):
    for i in range(2,n):
        if n % i == 0:
            return False
    return True

def solution(numbers):
    answer = 0
    
    arr = list(numbers)
    
    check = []
    
    for i in range(1,len(numbers)+1):
        for j in permutations(arr,i):
            s = ''
            for k in j:
                s += k
            
            if int(s) != 1:
                if int(s) !=0:
                    if int(s) not in check:
                        check.append(int(s))
                        if isTrue(int(s)):
                            answer+=1
                
            
        
    return answer

전형적인 그리디 문제이다.

해당 숫자로 구할 수 있는 수를 모두 구한 다음, 하나씩 소수인지 비교하면 끝나는 문제이다.

구할 수 있는 모든 수는 itertools 모듈을 이용하면 쉽게 구할 수 있다.