[ 완전 탐색 ] - 소수 찾기(Set, 재귀, Math.sqrt)

문제

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

각 종이 조각에 적힌 숫자가 적힌 문자열 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은 같은 숫자로 취급합니다.

풀이

1. 완전 탐색을 위해 재귀함수를 이용하여 모든 경우의 수를 체크하여 Set에 넣으면 중복은 제거된다.

2. Set에 들어간 숫자들을 순회하면서 소수 체크후 카운트

코드

package com.java.coding;

import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;

public class PrimeNumExam {
    static Set<Integer> set = new HashSet<>();
    
	public static void main(String[] args) {
		String numbers = "011";
		solution(numbers);
	}
    public static int solution(String numbers) {
    	// 1. 모든 조합 을 넣는다.
        recursive("",numbers);
        // 2. 소수 체크
        int count = 0;
        Iterator<Integer> it = set.iterator();
        while (it.hasNext()) {
            int number = it.next();
            if (isPrime(number))
                count++;
        }
        System.out.println(count);
        return count;
    }
    public static void recursive(String current, String numbers) {
        // 1. 숫자와 상관없이 다 넣는다. Set타입이라 중복 제거
        if (!current.equals(""))
        	set.add(Integer.valueOf(current));

        // 2. 남은 숫자 중 한 개를 더해 새로운 숫자를 만든다.
        for (int i = 0; i < numbers.length(); i++) {
        	System.out.println("current : "+current +numbers.charAt(i)+" : "+numbers.substring(0, i) + numbers.substring(i + 1));
            recursive(current + numbers.charAt(i), numbers.substring(0, i) + numbers.substring(i + 1));
        }
    }

    public static boolean isPrime(int n) {
        if(n==0 || n==1) {
            return false;
        }
        // 에라 토스 테네의 체
        for(int i=2;i<=(int)Math.sqrt(n);i++) {
            if(n % i == 0) {
                return false;
            }
        }
        return true;
    }
}

current : 0 : 11
current : 01 : 1
current : 011 : 
current : 01 : 1
current : 011 : 
current : 1 : 01
current : 10 : 1
current : 101 : 
current : 11 : 0
current : 110 : 
current : 1 : 01
current : 10 : 1
current : 101 : 
current : 11 : 0
current : 110 : 
2