[완전탐색] 프로그래머스 소수찾기 "JAVA"

🚀링크

https://programmers.co.kr/learn/courses/30/lessons/42839?language=java

💻문제

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

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

제한사항
numbers는 길이 1 이상 7 이하인 문자열입니다.
numbers는 0~9까지 숫자만으로 이루어져 있습니다.
"013"은 0, 1, 3 숫자가 적힌 종이 조각이 흩어져있다는 의미입니다.
입출력 예
numbers return
"17" 3
"011" 2
입출력 예 설명
예제 #1
[1, 7]으로는 소수 [7, 17, 71]를 만들 수 있습니다.

예제 #2
[0, 1, 1]으로는 소수 [11, 101]를 만들 수 있습니다.

11과 011은 같은 숫자로 취급합니다.

🌏문제풀이

재밌다.

👩‍💻코드

import java.util.*;

class Solution {
    boolean[] visited;
    ArrayList<Integer> all_card;
    
    public int solution(String numbers) {
        int answer = 0;
        int n = numbers.length();
        String[] card = new String[n];
        visited = new boolean[n+1];
        all_card = new ArrayList<Integer>();
        
        for(int i=0; i<n; i++){
            card[i] = Character.toString(numbers.charAt(i));
        }
        
        for(int r=0; r<n; r++){
            permutation(card, "", r+1, 0);
        }
        
        answer = all_card.size();
        
        return answer;
    }
    
    public void permutation(String[] card, String str, int r, int depth){
        if(depth == r){
            int tmp = Integer.parseInt(str);
            //소수이고 기존에 존재하지 않을 경우
            if(check(tmp)&&!all_card.contains(tmp)){
                all_card.add(tmp);
            }
            return;
        }
        
        for(int i=0; i<card.length; i++){
            if(!visited[i]){
                visited[i] = true;
                str = str + card[i];
                permutation(card, str, r, depth+1);
                //방문하지 않는 경우 붙여준 카드숫자를 다시 빼줘야함
                str = str.substring(0, str.length()-1);
                visited[i] = false;
            }
        }
    }
    
    public boolean check(int num){
        if(num == 0 || num == 1) return false;
        
        if(num == 2 || num == 3) return true;
        
        for(int i=2; i<Math.sqrt(num)+1; i++){
            if(num%i == 0) return false;
        }
        
        return true;
    }
}