[프로그래머스] 소수 찾기

문제

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

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

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

입출력 예

numbers	return
"17"	3
"011"	2

문제 출처

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

생각

• 만들 수 있는 모든 숫자들을 구해야 하기 때문에 완전탐색 -> DFS 로 풀면 되겠다!
• 각 숫자가 prime인지 판단하는 함수가 필요하겠다
• 중복해서 count하지 않으므로 set에다 집어넣고, 마지막에 set의 사이즈를 구하자

코드

import java.util.*;

class Solution {
    static Set<Integer> answer = new HashSet<>();
    static char[] nums;
    static int[] ch;

    public int solution(String numbers) {
        nums = numbers.toCharArray();
        ch = new int[numbers.length()];
        DFS("");
        return answer.size();
    }

    private void DFS(String curr) {
        if (!curr.equals("")) {
            int num = Integer.parseInt(curr);
            if (isPrime(num)) answer.add(num);
        }

        for (int i = 0; i < nums.length; i++) {
            if (ch[i] == 0) {
                ch[i] = 1;
                DFS(curr + nums[i]);
                ch[i] = 0; 
            }
        }
    }

    private boolean isPrime(int num) {
        if (num < 2) return false;
        for (int i = 2; i <= Math.sqrt(num); i++) {
            if (num % i == 0) return false;
        }
        return true;
    }
}

다른 사람의 풀이

• 아직 DFS가 익숙하지 않아서인지 다른 사람의 풀이들이 더 어렵게 느껴졌다,,,! 다른 사람의 풀이도 보고 익숙하게 해석할 수 있을 때까지 문제를 풀어봐야겠다 🤔🤔