[알고리즘] 백준 2023 파이썬 - 신기한 소수

 

2023번: 신기한 소수

 

문제

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

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

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

 

문제풀이

x=int(input())
#01
def sosu(n):
    if n==1:
        return False
    for i in range(2,int(n**0.5)+1):
        if(n%i==0):
            return False
    return True
#02
def dfs(num):
    if len(str(num))==x:
        print(num)
    else:
        for i in range(10):
            temp=num*10+i
            if sosu(temp):
                dfs(temp)
#03
dfs(2)  
dfs(3)
dfs(5)
dfs(7)

- #01 : 함수 sosu를 통해 n이 소수이면 True, 소수가 아니면 False를 반환한다.

- #02 : 함수 dfs를 통해 num이 길이 x가 되면 출력해준다.

           그 외 경우는 0~9까지 i를 for문으로 입력받으며 백트래킹을 통해

           왼쪽부터 1자리, 2자리, 3자리 .. 이렇게 모두 만족하여 num의 길이가 x가 되면 출력해준다.

- #03 : 애초에 왼쪽 1자리부터 소수여야된다.

           1~9까지 중에 소수는 2,3,5,7 밖에 없으니 위와 같이 작성한다.