Language/Python

[Python / 프로그래머스 level 2] 소수 찾기

ej503 2021. 8. 18. 09:24

from itertools import permutations
def solution(numbers):
    num_list = []
    for i in range(1, len(numbers)+1):
        test_list = permutations(numbers, i)
        for j in test_list:
            num_list.append(int("".join(j)))
    num_list = set(num_list)
    if 0 in num_list:
        num_list.remove(0)
    if 1 in num_list:
        num_list.remove(1)
    answer = len(num_list)
    for i in num_list:
        if i != 2:
            for j in range(2, int(i**0.5)+1):
                if i % j == 0:
                    answer -=1
                    break
    return answer