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

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/42839

프로그래머스

 

문제 설명

각 종이 조각에 적힌 숫자가 적힌 문자열 numbers가 주어졌을 때, 종이 조각으로 만들 수 있는 소수가 몇 개인지 return 하도록 함수를 작성

제한 사항

• numbers는 길이 1 이상 7 이하인 문자열입니다.
• numbers는 0~9까지 숫자만으로 이루어져 있습니다.
• "013"은 0, 1, 3 숫자가 적힌 종이 조각이 흩어져있다는 의미입니다.

 

풀이

1. numbers의 순열을 구한다.
2. SET을 이용해, 각 순열로 이루어진 숫자의 중복을 제거한다.
3. 소수를 찾는 함수를 통해 소수를 찾는다.

 

✅ Code - 성공

from itertools import permutations

# 소수 찾는 함수
def find_prime(n):
    if n < 2:
        return False
    
    for i in range(2, int(n**(1/2)) + 1):
        if n % i == 0:
            return False
        
    return True
        
def solution(numbers):
    answer = 0
    permu = []
    for i in range(1, len(numbers) + 1):
        permu += list(permutations(numbers, i))
    
    num_lst = set()
    for c in permu:
        c = int(''.join(c))
        num_lst.add(c)
    
    for n in num_lst:
        if (find_prime(n)):
            answer += 1
    
    return answer