[백준/Python] 2023번 신기한 소수

1️⃣문제 설명

수빈이가 세상에서 가장 좋아하는 것은 소수이고, 취미는 소수를 가지고 노는 것이다. 요즘 수빈이가 가장 관심있어 하는 소수는 7331이다.

7331은 소수인데, 신기하게도 733도 소수이고, 73도 소수이고, 7도 소수이다. 즉, 왼쪽부터 1자리, 2자리, 3자리, 4자리 수 모두 소수이다! 수빈이는 이런 숫자를 신기한 소수라고 이름 붙였다.

수빈이는 N자리의 숫자 중에서 어떤 수들이 신기한 소수인지 궁금해졌다. N이 주어졌을 때, 수빈이를 위해 N자리 신기한 소수를 모두 찾아보자.

2️⃣입력

첫째 줄에 N(1 ≤ N ≤ 8)이 주어진다.

3️⃣출력

N자리 수 중에서 신기한 소수를 오름차순으로 정렬해서 한 줄에 하나씩 출력한다.

---

👩🏻‍💻작성한 코드

import sys

sys.setrecursionlimit(10 ** 6)

n = int(input())	# 몇자리 수인지 입력


# '에라토스테네스의 체'로 소수 판별
def Eratos(checkPrime):
    for i in range(2, int(checkPrime ** 0.5) + 1):
        if int(checkPrime) % i == 0:
            return False
    return True


def DFS(num):
    if len(str(num)) == n:
        print(num)
    else:
        for i in range(10):
            temp = num * 10 + i
            if Eratos(temp):
                DFS(temp)


DFS(2)
DFS(3)
DFS(5)
DFS(7)

 

💡코드 설명

에라토스테네스의 체로 소수 판별을 하는 Eratos() 함수를 만들었다. 

 

DFS() 함수에서는 매개변수로 받은 숫자의 길이가 n자리일 때, 조건을 모두 충족하므로 해당 값을 반환하도록 하고, 아닐 때는 숫자에 10을 곱하여 자릿수를 늘려주고 0~9까지의 수를 차례로 더해줘서 새로 만들어진 수인 temp가 소수인지 아닌지 판별한다. 

 

그리고 temp가 소수이면 DFS를 다시 실행한다. -> 재귀함수 실행

 

왼쪽부터 한자리수일 때 소수여야 하므로 한자리 소수인 2, 3, 5, 7을 DFS() 함수에 넣어주면 원하는 값들을 얻을 수 있다.