[프로그래머스] LV.2 소수 찾기

[프로그래머스] LV.2 소수 찾기

 

1. 문제 : Link

문자열로 숫자가 주어지고 순열 이용해서 소수찾기

 

2. 풀이

순열이용 중복삭제 및 제곱근 이용해서 소수찾기

 

3. 코드

from itertools import permutations
import math

def solution(numbers):
    answer = 0
    num = []
    
    for i in range(len(numbers)):
        a = list(permutations(numbers, i+1))
        num = num + a
    
    num2 = []
    for i in num:
        num2.append(int(''.join(k for k in i)))
    
    num2 = list(set(num2))
    
    for n in num2:
        cnt = 0
        sqrt =  int(math.sqrt(n))
        if n == 1 or n == 0:
            continue
        for i in range(2, sqrt+1):
            if n % i == 0:
                cnt += 1
        if cnt == 0:
            answer += 1 
    return answer