[프로그래머스 Lv.2] 소수 찾기 (Python)

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/42839#

프로그래머스

풀이

문자열을 이루는 각 숫자를 조합하여 특정 숫자를 만들고 이 수가 소수인지 체크하면 되는 간단한 문제이다.

풀이 순서는 다음과 같다.

1. 문자열을 한 자리씩 잘라서 한 자리의 정수로 만든다.
2. 한 자리 정수들을 사용하여 만들 수 있는 수들을 permutation을 사용해 모두 알아낸다.
3. 2번에서 만든 수들이 소수인지 체크하고 소수의 수를 기록한다.

from itertools import permutations

def solution(numbers):
    _numbers = []
    for i in numbers: _numbers.append(i) # 문자열을 한 자리씩 잘라서 저장
    
    number_set = set() # 중복방지를 위해 set 자료형 사용
    for i in range(1, len(numbers) + 1):
        nums = list(permutations(_numbers, i)) # permutation을 사용하여 순열을 통해 숫자 조합 생성
        for num in nums: number_set.add(int("".join(num)))
    
    cnt = 0
    for num in number_set: # 순열로 조합한 숫자들을 하나씩 순회하며
        if check_if_prime(num): # 해당 숫자가 소수라면
            cnt += 1
                    
    return cnt


def check_if_prime(num):
    if num < 2: return False
    
    for i in range(2, int(num ** 0.5) + 1): # 소수 판별
        if num % i == 0: return False
    return True