[level 2] 소수 찾기

1. 문제

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

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

 

제한사항

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

 

2. 문제 풀이

1) 각 자릿수 별로 순열을 구한다(순서에 따라서 숫자가 달라지므로 조합보단 순열)

2) 만약 num_list에 해당 숫자가 없다면, 리스트에 구한 숫자를 추가한다.

3) 위에서 구한 숫자 조합 리스트로 소수 판별을 진행한다. 소수가 맞으면 answer에 +1을 한다. 

from itertools import permutations
def solution(numbers): 
    num_list = []
    answer = 0
    
    for i in range(1, len(numbers)+1):
        for perm in permutations(numbers, i):
            num = int(''.join(perm))
            if num not in num_list:
                num_list.append(num)
                
    for num in num_list:
        
        if num < 2:
            continue
        
        for i in range(2, int(num**0.5)+1):
            if num % i == 0:
                break
        else:
            answer +=1 
    
    
    return answer

 

자릿수별로 어떻게 조합을 구하는지만 생각해내면 어려운 문제는 아니었던 것 같다!