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

링크텍스트

문제 이해

한자리 숫자가 적힌 종이 조각이 흩어져있습니다. 흩어진 종이 조각을 붙여 소수를 몇 개 만들 수 있는지 알아내려 합니다.
각 종이 조각에 적힌 숫자가 적힌 문자열 numbers가 주어졌을 때, 종이 조각으로 만들 수 있는 소수가 몇 개인지 return 하도록 solution 함수를 완성해주세요.
제한사항
numbers는 길이 1 이상 7 이하인 문자열입니다.
numbers는 0~9까지 숫자만으로 이루어져 있습니다.
"013"은 0, 1, 3 숫자가 적힌 종이 조각이 흩어져있다는 의미입니다.

나만의 이해

스트링을 우선 리스트로 변환해서 조합을 찾고 그 조합이 소수인지 판단하는 문제

Python

from itertools import permutations
def check(num):
    if num <=1:
        return False

    for i in range(2,int(num**(1/2))+1):
        if num % i ==0:
            
            return False
    return True

def solution(numbers):
    if len(numbers) ==1:
        if check(int(numbers)):
            return 1
        return 0
    items = [x for x in numbers]
    ans =[]
    ans.extend(items)
    for i in range(2,len(numbers)+1):
        ans.extend(list(map(''.join, permutations(items,i))))
    ans = [x for x in ans if x !='0']
    ans = [int(x) if x[0] != "0"  else int(x[1:]) for x in ans]
    ans = list(set(ans))
    ans.sort()
    
    count =0

    for ele in ans:
        if check(ele):
            print(ele)
            count +=1
    return count

check는 소수 판별하는 방법 (자신의 루트까지 %해보면 나옴).
여기서 중요한것은 permutations 사용법이랑 ans 판단을 잘해야된다.