[Programmers] 소수 찾기

문제

한자리 숫자가 적힌 종이 조각이 흩어져있습니다. 흩어진 종이 조각을 붙여 소수를 몇 개 만들 수 있는지 알아내려 합니다.

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

제한사항

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

입출력 예 설명

 

예제 #1
[1, 7]으로는 소수 [7, 17, 71]를 만들 수 있습니다.

예제 #2
[0, 1, 1]으로는 소수 [11, 101]를 만들 수 있습니다.

• 11과 011은 같은 숫자로 취급합니다

---

풀이

풀이방법

 

1. 문자열을 리스트로 변환

2. 조합할 숫자를 1부터 문자열 길이까지 반복

3. itertools의 permutations를 사용해 글자의 순열 생성

4. 순열을 문자열로, 문자열을 정수로 바꾼 뒤, 처음 확인하는 수이면, 소수 판별

5. 소수가 맞으면 개수 1 증가

from itertools import permutations
from math import sqrt

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

def solution(numbers):
    answer = 0
    lst = list(numbers)
    tmp = []
    
    for i in range(1, len(numbers)+1):        
        for i in permutations(lst, i):
            t = int("".join(list(i)))
            if t not in tmp:
                tmp.append(t)
                if isprime(t):
                    answer+=1
    
    return answer