[프로그래머스] 소수 찾기 / 완전 탐색(Brute-force Search)

이미지를 클릭하면 문제 사이트로 이동합니다.

 

 

 

💭 문제 이해

 

한 자리의 숫자들이 나열된 문자열을 조합해서 만들 수 있는 소수의 갯수를 구하는 문제이다.

 

파이썬의 기본 모듈인 itertools.permutations를 이용하여 모든 경우를 구하고,

소수 판별은 숫자 범위가 0~9로 적으므로 2~n-1까지 나눠보면서 직접 찾아냈다.

 

 

 

 

구현 언어: Python

 

from itertools import permutations

# 소수 판별 함수
def is_prime_number(num: int) -> bool:
    if num < 2:
        return False
    for i in range(2, num):
        if num % i == 0:
            return False
    return True


def solution(numbers):
    answer = set()  # 중복되지 않는 모든 경우를 저장할 set
    for i in range(1, len(numbers)+1):
        num_list = list(''.join(tup) for tup in permutations(numbers, i))
        for num in num_list:
            if is_prime_number(int(num)):
                answer.add(int(num))
    return len(answer)