[프로그래머스 Lv.1] 소수 찾기 (완전 탐색)

[프로그래머스 Lv.1] 소수 찾기 (완전 탐색)

파이썬 풀이

from itertools import permutations
import math

def solution(numbers):
    answer = 0
    s = set()
    for n in range(1, len(numbers) + 1):
        for p in list(permutations(numbers, n)):
            check = 0
            num = ''
            
            for k in range(n):
                num += p[k]
            num = int(num)
            
            if num < 2:
                check = -1
            else:
                for i in range(2, int(math.sqrt(num)) + 1):
                    if num % i == 0:
                        check = -1
                        break
                        
            if check != -1:
                s.add(num)  
                
    return len(s)

• permutations를 이용해서 모든 조합을 구함