취준시절/프로그래머스

[프로그래머스] 소수찾기 - Python

MAYMIN 2021. 8. 26. 20:38
728x90
SMALL

permutations 의 활용이 정말 중요했음 !

증말 유용한 내장함수 ... !! 잊지말기!!

저번에도 썼었는데 까먹음 ㅎㅎ...

순열만 잘 만들어놓으면 소수만 찾으면 되는 문제 !!

from itertools import permutations

def solution(numbers):
    answer = list()
    numlist = [i for i in numbers]
    arr = list()
    for i in range(1, len(numlist) + 1):
        arr += list(permutations(numlist, i))
    arrangelist = [int(i) for i in list(map(''.join, arr))]

    for i in arrangelist:
        if i < 2:
            continue
        check = True
        for j in range(2, i):
            if i % j == 0:
                check = False
                break
        if check == True:
            answer.append(i)

    answer = list(set(answer))
    return len(answer)
728x90
LIST