문제 설명

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

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

제한 사항

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

입출력 예제

numbers return
“17” 3
“011” 2

문제풀이

from itertools import permutations
from math import sqrt

def all_case_permutations(number):
    result = []
    for k in range(1, len(number)+1):
        tmp = list(map(int,map(''.join, permutations(number, k))))
        result += tmp
    return list(set(result))

def count_prime_number(per_list):
    count = 0
    for n in per_list:
        if is_prime_number(n):
            count+=1
    return count

def is_prime_number(number):
    if number==0 or number==1:
        return False
    else:
        for i in range(2, int(sqrt(number)) + 1):
            if number % i == 0:
                return False
        return True       
        
def solution(number):
    return count_prime_number(all_case_permutations(number))

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