한자리 숫자가 적힌 종이 조각이 흩어져있습니다. 흩어진 종이 조각을 붙여 소수를 몇 개 만들 수 있는지 알아내려 합니다.
각 종이 조각에 적힌 숫자가 적힌 문자열 numbers가 주어졌을 때, 종이 조각으로 만들 수 있는 소수가 몇 개인지 return 하도록 solution 함수를 완성해주세요.
numbers | return |
---|---|
17 | 3 |
011 | 2 |
from itertools import permutations
def check(x):
if x<2:
return False
for i in range(2, x):
if x % i == 0:
return False
return True
def solution(numbers):
# 집합에 permuation 통해 탐색하며 값 추가
answer=[]
per=[]
for i in range(1,len(numbers)+1):
per=list(map(''.join,permutations(numbers,i)))
per=set(per)
for p in per:
if check(int(p)):
answer.append(int(p))
answer=set(answer)
return len(answer)
문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/42839