💡 문제

소수 찾기

• 유형 완전탐색

• 문제 설명 한자리 숫자가 적힌 종이 조각이 흩어져있습니다. 흩어진 종이 조각을 붙여 소수를 몇 개 만들 수 있는지 알아내려 합니다. 각 종이 조각에 적힌 숫자가 적힌 문자열 numbers가 주어졌을 때, 종이 조각으로 만들 수 있는 소수가 몇 개인지 return 하도록 solution 함수를 완성해주세요.

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

• 입력
  • ex1) "17"
  • ex2) "011

• 출력
  • ex1) 3
  • ex2) 2

✍🏻 풀이

SOL 1

# SOL 1) 

from itertools import permutations

def solution(numbers):
    result = set()
    for k in range(1, len(numbers)+1):
        for i in permutations(numbers, k):
            result.add(int(''.join(list(i))))

    cnt = 0
    for i in result:
        if i in (2, 3):
            cnt += 1
        temp = 0
        for j in range(2, int(i**0.5)+1):
            if i % j == 0:
                temp = 1
                break
            elif (j == int(i**0.5)) and temp == 0:
                cnt += 1        
    return cnt

🌵 정리

깃헙에서 전체 코드 확인하기