프로그래머스_소수 찾기(JAVA)

https://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

입출력 예 설명

예제 #1
[1, 7]으로는 소수 [7, 17, 71]를 만들 수 있습니다.

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

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

[코드]

import java.util.*;
class Solution {
    ArrayList<Integer> candidate = new ArrayList<>();    //종이 조각으로 만들 수 있는 모든 수를 저장할 Arraylist
    boolean[] visited;    //종이 조각으로 만들 수 있는 수 구할 때 해당 숫자 포함 여부
    String str="";

//종이 조각으로 만들 수 있는 모든 수 구하기
    public void combination(ArrayList<Integer> al, int n, int count){
        if(count==n){
            if(!candidate.contains(Integer.parseInt(str)))   
                candidate.add(Integer.parseInt(str));    //중복 막기 위해 arraylist인 candidate 내에 해당 수의 존재여부 파악 후 없으면 arraylist에 넣어줌
            return;
        }
        for(int i=0;i<al.size();i++){
            if(!visited[i]){
                if(al.get(i)==0){   //현재 수가 0이고 str에 들어있는 수가 하나도 없을 때, 0이 맨 처음 오는 것을 막기 위함 (ex) 011 -> 11) 
                    if(str.length()>0){ 
                        visited[i]=true;
                        str+=Integer.toString(al.get(i));
                        combination(al, n, count+1);
                        visited[i]=false;
                        str= str.substring(0,str.length()-(Integer.toString(al.get(i))).length());
                    }
                }
                else{
                    visited[i]=true;
                    str+=Integer.toString(al.get(i));
                    combination(al, n, count+1);
                    visited[i]=false;
                    str = str.substring(0,str.length()-(Integer.toString(al.get(i))).length());
                }
                
            }
        }
    }

//소수 판별
    public int primeNum(int num){
        if(num==1)
            return 0;
        
        for(int i=2;i<num;i++){
            if(num==2)
                return 1;
            if(num%i==0)    
                return 0;
            
        }
        return 1;
    }
    public int solution(String numbers) {
        int answer = 0;
        int n = Integer.parseInt(numbers);   //문자열 numbers를 정수로 변환
        ArrayList<Integer> al = new ArrayList<>();   
       

        //n을 한 숫자씩 쪼개서 넣어줌
        for(int i=0;i<numbers.length();i++){
            al.add(n%10]);
            n/=10;
        }
        visited = new boolean[al.size()];
        
        //조합 생성
        for(int i=1;i<=al.size();i++)
            combination(al, i, 0);
        
        //소수판별
        for(int i=0;i<candidate.size();i++)
            answer+=primeNum(candidate.get(i));
      
        return answer;
    }
}