[BOJ][🟡5][백준#02023] 신기한 소수

문제 출처

BAEKJOON Online Judge #2023

문제

수빈이가 세상에서 가장 좋아하는 것은 소수이고, 취미는 소수를 가지고 노는 것이다. 요즘 수빈이가 가장 관심있어 하는 소수는 7331이다. 7331은 소수인데, 신기하게도 733도 소수이고, 73도 소수이고, 7도 소수이다. 즉, 왼쪽부터 1자리, 2자리, 3자리, 4자리 수 모두 소수이다! 수빈이는 이런 숫자를 신기한 소수라고 이름 붙였다. 수빈이는 N자리의 숫자 중에서 어떤 수들이 신기한 소수인지 궁금해졌다. N이 주어졌을 때, 수빈이를 위해 N자리 신기한 소수를 모두 찾아보자.

입력

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

출력

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

예제

입력

4

출력

2333
2339
2393
2399
2939
3119
3137
3733
3739
3793
3797
5939
7193
7331
7333
7393

My Sol

def is_prime(n):
    for i in range(2, int(n**(1/2))+1):
        if not n%i: return False
    return True

N = int(input())
now = 1
primes = [2, 3, 5, 7]
while now < N:
    new_primes = []
    for prime in primes:
        for i in range(1, 10):
            p = prime*10+i
            if is_prime(p):
                new_primes.append(p)
    primes = new_primes
    now += 1

for p in primes:
    print(p)

소수 판정을 위한 에라토스테네스의 체 함수를 사용하는 문제였다.

1. 1자리 소수 리스트를 준비한다.
2. 해당 소수를 10씩 곱하고 1부터 9까지 더한 값들을 소수판정하여 소수인 수들만 primes에 다시 담는다.
3. now가 N이 될 때까지 반복한다.

for문의 순서를 prime을 먼저 두어야 작은 수부터 자연스럽게 배치된다.

결과

맞았습니다!!