프로그래머스 - 소수 찾기 - DFS - Java

1. 문제

https://school.programmers.co.kr/learn/courses/30/lessons/42839

한자리 숫자가 적힌 종이 조각이 흩어져있습니다. 흩어진 종이 조각을 붙여 소수를 몇 개 만들 수 있는지 알아내려 합니다.

각 종이 조각에 적힌 숫자가 적힌 문자열 numbers가 주어졌을 때, 종이 조각으로 만들 수 있는 소수가 몇 개인지 return 하도록 solution 함수를 완성해주세요.

• numbers는 길이 1 이상 7 이하인 문자열입니다.
• numbers는 0~9까지 숫자만으로 이루어져 있습니다.
• "013"은 0, 1, 3 숫자가 적힌 종이 조각이 흩어져있다는 의미입니다.

numbers	return
"17"	3
"011"	2

2. 풀이

참고 풀이

1. 각 종이로 만들 수 있는 조합을 DFS로 구한다. 단 중복을 제거하기 위해 Set을 선언해서 넣어준다.

isPrime()함수를 이용해 Set에 넣은 숫자들이 소수인지 확인한다.

✨핵심 Point

소수 판별 코드

public boolean isPrime(int n) {
	
	if(n <= 1) {
		return false;
	}
	
	for(int i = 2; i * i <= n; i++) {
		if(n % i == 0) {
			return false;
		}
	}
	return true;
}

3. 전체코드

import java.util.*;

class Solution {
    
    static HashSet<Integer> set = new HashSet<>();
    static boolean[] visit = new boolean[7];
    
    public int solution(String numbers) {
        dfs(numbers, "", 0);    
        
        int count = 0;
        for(Integer num : set){
            if(isPrime(num)){
                count++;
            }
        }
        return count;
    }
    
    public void dfs(String numbers, String s, int depth){
        if(depth > numbers.length()){
            return;
        }
        
        for(int i = 0; i < numbers.length(); i++){
            if(!visit[i]){
                visit[i] = true;
                set.add(Integer.parseInt(s + numbers.charAt(i)));
                dfs(numbers, s + numbers.charAt(i), depth+1);
                visit[i] = false;
            }
        }
    }
    
    public boolean isPrime(int n){
        if(n <= 1){
            return false;
        }
        
        for(int i = 2; i * i <= n; i++){
            if(n % i == 0){
                return false;
            }
        }
        return true;
    }
}