본문 바로가기
프로그래머스/코딩테스트 고득점 kit

프로그래머스 완전탐색 LV2 - 소수 찾기

by 맴썰 2021. 10. 6.

문제 설명

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

각 종이 조각에 적힌 숫자가 적힌 문자열 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 {
    public int solution(String numbers) {
        int answer = 0;
        ArrayList<Character> abc = new ArrayList<>();
        char[] str = numbers.toCharArray();
        Arrays.sort(str);
        for(int i=0; i<str.length; i++) abc.add(str[i]);
        String temp = String.valueOf(str);
        StringBuffer sb = new StringBuffer(temp);
        temp = sb.reverse().toString();//내림차순정렬 -> 주어진수의 최댓값
        int[] era = new int[Integer.parseInt(temp)+1]; // 최댓값만큼의 개수의 배열(에라토스테네스의 체)
        era[0] = 1; era[1] = 1;
        String[] ps = new String[era.length];
        int idx=0;
        for(int i=2; i<era.length; i++){
            if(era[i]==0){
                ps[idx++] = String.valueOf(i);
                for(int j=i+i; j<era.length; j+=i){
                    era[j] = 1;
                    
                }
            }
            else continue;
        }int cnt=0;
        for(int i=0; i<ps.length; i++){if(ps[i]!=null)cnt++;}
        String[] ss = new String[cnt];
        for(int i=0; i<ss.length; i++) ss[i] = ps[i]; //최댓값 이하의 소수값들을 가진 배열
        boolean no = false;  char[] a = new char[str.length];
        for(int i=0; i<ss.length; i++){
            char[] temparr = ss[i].toCharArray(); no = false;
            char[] copy  = refresh(str,a); 
            for(int j=0; j<temparr.length; j++){ 
                for(int k=0; k<copy.length; k++){ 
                    if(temparr[j]==copy[k]){ //주어진 수를 포함하면 
                        copy[k] = 'a';  //재사용 방지 변환
                        temparr[j] = 'b'; //재사용 방지 변환
                        break;
                    }
                    else if(!abc.contains(temparr[j])){no = true; break;} //주어진 수가 아닌 수를 가지고 있으면 Break
                }
                if(no) break;
            }
            String tempstr = String.valueOf(temparr);
            tempstr = tempstr.replace("b",""); // 정답으로 인정된 소수 중 변환되지 않은 수 찾기
            if(!no&&tempstr.equals("")) {answer++;}
        }
        return answer;
    }
    char[] refresh(char[] a, char[] b){
        for(int i=0; i<a.length; i++){
            b[i] = a[i];
        }
        return b;
    }
}