백준 2023 신기한 소수 찾기

문제

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

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

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

입력

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

출력

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

예제 입력 1
4
예제 출력 1
2333
2339
2393
2399
2939
3119
3137
3733
3739
3793
3797
5939
7193
7331
7333
7393

나의 풀이

import java.util.Scanner;

public class BJ_2023_신기한소수찾기 {
    static int N;

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        N = sc.nextInt();
        DFS(2,1);
        DFS(3,1);
        DFS(5,1);
        DFS(7,1);
    }
    static boolean isPrime(int num){
        for(int i = 2; i <= Math.sqrt(num); i++){
            if(num % i == 0){
                return false;
            }
        }
        return true;
    }

    static void DFS(int num, int digit){
        if(digit == N){
            System.out.println(num);
            return;
        }
        for(int i = 1; i< 10; i++){
            if(isPrime(num * 10 + i)){
                DFS(num * 10 + i, digit+1);
            }
        }
    }
}

돌아보기

한 번 풀었던 문제를 다시 못 푼다는 것은 그만큼 문제가 있다는 것이겠지
코딩테스트가 가까이 왔다고 하더라도 그것에 얽매이지 말고
효율성 정확성 등등 제대로 따져가면서 고민하고 문제를 풀고 정리하도록 하자
위 풀이처럼 첫 자리가 소수인 수만 가지고 dfs를 돌리는 것이
2부터 9까지 먼저 소수 확정을 한 후 돌리는 것보다 시간이 약간 빠르게 나왔다.
소수를 구하는 방법에는 여러가지가 있고 가장 유명한 것은 '에라토스테네스의 체'가 있지만
그 부분은 다음에 정리하도록 하자

출처 및 참고 : 백준, Do it! 코딩테스트 자바 편