[프로그래머스] 소수 찾기 -Java

코딩테스트 문제를 공부하다보면 응용 할 수 있는 좋은 문제들이 나온다
해당 문제는 재귀를 이용한 숫자카드의 모든 경우의 수를 찾고, '에라토스테네스의 체' 를 이용하여 소수인지 판별하면 된다.

1. numbers에 들어있는 숫자카드를 중복없이 만들 수 있는 모든 경우의 수를 찾는다.
2. 각각 숫자카드를 확인하여 소수인지 판단한다.

아래의 코드로 구현할 수 있다.

import java.util.*;

class Solution {
	// set을 이용하여 중복 제거
    static HashSet<Integer> map = new HashSet<>();
    
    // 재귀함수를 호출하여 숫자카드의 모든 경우의 수를 set에 담는다
    public static void recrusive(String a, String b) {
        if(!"".equals(a)) {
			map.add(Integer.parseInt(a));
		}
		
		for(int i = 0 ; i < b.length(); i++) {
			recrusive(a + b.charAt(i), b.substring(0, i) + b.substring(i + 1));
		}
    }
    
    // 소수인지 확인하고, 합산하여 리턴해준다.
    public static int sqrt(HashSet<Integer> map, int answer) {
        Iterator<Integer> iter = map.iterator();
        
        while(iter.hasNext()) {
            boolean result = true;
            int val = iter.next();

            if(val == 0 || val == 1) {
                continue;
            }

            int root = (int) Math.sqrt(val);
            for(int i = 2; i <= root; i++) {
                if(val % i == 0) {
                    result = false;
                    break;
                }
            }

            if(result) {
                answer++;
            }

        }
        
        return answer;
    }
    
    public int solution(String numbers) {
        int answer = 0;
		// 탐색 시작
        recrusive("", numbers);
        
        return sqrt(map, answer);
    }
}