[백준] G5 2023번 신기한 소수 (JAVA)

문제 출처 - Baekjoon Online Judge

문제는 여기

2023번: 신기한 소수

[문제] 

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

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

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

[입력]

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

[출력]

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

 

 

---

[풀이]

1. 신기한 소수인지를 체크하기 위해서 prime 체크를 하는 함수를 만들어준다.

2. 신기한 소수를 만들기 위해 DFS를 사용해서 체크해준다.

3. DFS를 사용해 소수인 경우, 깊이를 증가시키고 아니면 return 하는 방식으로 구현하였다.

[접근]

1. 처음에는 에라토스테네스의 체를 사용해서 소수 배열을 만들어두고, 신기한 소수인지 체크를 하려고 하였다.

2. 이렇게 하니 메모리 초과가 발생했다.

3. 검색을 통해 배열을 만드는 것이 아닌 바로 소수인지 체크하는 방식을 사용하면 된다는 것을 알게 되었다.

4. 소수인지 체크하는 함수를 만들어 주었다.

5. 자릿수를 올려가면서 소수인지 체크를 해주어야 하는데 이를 DFS를 사용해서 구현해주었다.

[코드]

package BOJ_gold;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main_G5_2023 {
	static int n;

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		n = Integer.parseInt(br.readLine());
		
		// dfs 탐색
		dfs("", 0);
	}
	
	private static void dfs(String s, int len) {
		// n자리수라면
		if (len == n) {
			System.out.println(s);
			return;
		}
		// 1~9까지 체크 (0은 생략해도 됨 - 0으로 끝나면 5의 배수로 무조건 소수가 불가능)
		for(int i = 1; i <= 9; i++) {
			// 소수 체크해서 다음 탐색
			if(prime(Integer.parseInt(s + i))) {
				dfs(s + i, len + 1);
			}
		}
	}
	
	// 소수 체크
	private static boolean prime(int num) {
		// 1이면 소수가 아님
		if(num == 1)
			return false;
		
		int sqrt = (int)Math.sqrt(num);
		
		for(int i = 2; i <= sqrt; i++) {
			// 나눠서 떨어지면 소수가 아님
			if(num % i == 0)
				return false;
		}
		// 아니라면 소수
		return true;
	}
}