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

코딩테스트 연습 - 소수 찾기

import sys
from itertools import permutations

input = sys.stdin.readline

numbers = input().rstrip()


def isPrime(n):
    if n < 2:
        return False
    for i in range(2, int(n ** 0.5) + 1):
        if n % i == 0:
            return False
    return True


def solution(numbers):
    answer = []
    for i in range(len(numbers)):
        temp = list(set((map(int, map("".join, permutations(numbers, i + 1))))))
        print(temp)
        for n in temp:
            if isPrime(n):
                answer.append(n)
    return len(set(answer))


print(solution(numbers))

 

해설

permutations를 통해 모든 조합을 추출하고 set으로 중복되는 값을 제거했다.

그렇게 얻는 데이터를 isPrime 함수를 통해 소수 판정을 한 뒤, 소수이면 answer에 추가한다.

answer의 중복 값을 제거하기 위해 set을 이용하고 answer에 길이를 리턴한다.