링크

소수 찾기

문제 설명

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

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

제한 사항

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

입출력 예 설명

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

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

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

풀이

이경우에는 {1,7}, {0,1,1}의 각각의 조합을 맞춰줘야 하기 때문에 백트래킹을 사용하면 된다. 여기서 중요한 점은 hashset을 사용하여야 한다. 그러한 이유는 011도 11로 간주하기 때문에 StringBuilder값에 011을 넣어놓고 Integer.parseInt(011)를 한다면 11로 숫자가 변환되기에 StringBuilder와 HashSet을 이용한다.

Code

import java.util.*;

class Solution {
    static HashSet<Integer> set=new HashSet<Integer>();
    static boolean visited[];
    public int solution(String numbers) {
        int answer=0;
        visited=new boolean[numbers.length()];
        StringBuilder sb=new StringBuilder();
        
        for(int i=0; i<numbers.length(); i++)
            DFS(sb, numbers, 0);
        
        for(int number : set)
            if(isPrimeNumber(number)) answer++;
        
        return answer;
    }
    
    public static void DFS(StringBuilder sb, String numbers, int depth)
    {
        if(sb.length()!=0)
            set.add(Integer.parseInt(sb.toString()));
        
        if(depth==numbers.length())
            return;
        
        for(int i=0; i<numbers.length(); i++)
        {
            if(!visited[i])
            {
                visited[i]=true;
                sb.append(numbers.charAt(i));
                DFS(sb, numbers, depth+1);
                visited[i]=false;
                sb.setLength(sb.length()-1);
            }
        }
    }
    
    public static boolean isPrimeNumber(int number)
    {
        if(number<=1) return false;
        
        for(int i=2; i<=Math.sqrt(number); i++)
            if(number%i==0) return false;
        return true;
    }
}