[파이썬] 프로그래머스 소수 찾기

순열을 이용해서 비교적 쉽게 풀었다. 시간 초과가 안나는게 신기하다.

 

 

from itertools import permutations
 
def sosu(n):
    if n<=1:
        return False
 
    else:
        for i in range(2,int(n//2)+1):
            if n%i == 0:
                return False
 
        return True
 
 
 
def solution(numbers):
    answer = 0
 
    temp = []
 
    n = list(numbers)
 
    for i in range(1,len(n)+1):
        k = list(permutations(list(numbers),i))
 
        for i in range(0,len(k)):
            num = int("".join(k[i]))
            if num not in temp:
                temp.append(num)
 
    for i in range(0,len(temp)):
        if sosu(temp[i]) == True:
            answer+=1
 
    return answer