💻코딩테스트 문제 풀이5

Chapter6. 완전탐색

[문제22] 소수찾기 - Level2

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

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

- numbers는 길이 1 이상 7 이하인 문자열입니다.

- numbers는 0~9까지 숫자만으로 이루어져 있습니다.

- "013"은 0, 1, 3 숫자가 적힌 종이 조각이 흩어져 있다는 의미입니다.

• 1. 소수인지 판별하는 함수를 만듭니다.

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

• 2. 주어진 종이 조각으로 만들 수 있는 모든 숫자를 찾습니다.

from itertools import permutations

for i in range(1, len(numbers) + 1):
	num.append(list(permutations(numbers, i)))
num = [int(''.join(y)) for x in num for y in x]

• 3. 1번과 2번을 사용하여 나온 소수 배열의 길이를 계산합니다.

for i in num:
	if checkPrime(i):
    	answer.append(i)
        
return len(set(answer))

//최종 정답//
from itertools import permutations

def checPrime(n):
	if n < 2:
    	return False
        
	for i in range(2, int(n ** 0.5) + 1):
    	if n % i == 0:
        	return False
    return True
    
def solution(numbers):
	answer = []
    numbers = list(numbers)
    #prime = primes(10 ** len(numbers) - 1)
    num = []
    
    for i in range(1, len(numbers) + 1):
    	num.append(list(permutations(numbers, i))
    num = [int(''.join(y)) for x in num for y in x]
    
    for i in num:
    	if checkPrime(i):
        	answer.append(i)
            
    return len(set(answer))