프로그래머스 - 소수 찾기(level2)

programmers.co.kr/learn/courses/30/lessons/42839

코딩테스트 연습 - 소수 찾기

 

1. 유형

순열, 수학

 

2. 접근법

• 순열
• 소수 판단
• 011같은 수 예외처리나 중복체크

3. 풀이

- 종이조각을 붙여서 다양한 경우를 만들어야 한다. 이때, 순서도 상관이 있으므로 순열을 구한다.

 

일반적으로 길이가 고정된 순열이 아니다. 길이가 유동적인 순열이므로 이점이 다른 순열문제와 차별점이라 생각한다.

 

그러므로 조합+순열 느낌이 나는 문제였다.

 

- 소수판단은 2~ 제곱근까지만 판단하면 된다.

 

- 011 같은 예외처리는 String을 Integer.valueOf로 고치면 알아서 처리된다.

 

- 소수 판단이 끝나면 HashSet<Integer> 자료구조에 넣어주면 중복처리가 끝난다. 그 후, answer에는 HashSet의 길이를 넣어준다.

 

4. 코드

import java.util.*;

class Solution {
    static String str;
    static HashSet<Integer> diction;
    public int solution(String numbers) {
        int answer = 0;
        int len = numbers.length();
        str=numbers;
        diction = new HashSet<>();
        boolean visit[] = new boolean[len];
        for(int i=1; i<=len; i++){
            perm_dfs(0, i, "", visit);
        }
        answer = diction.size();
        return answer;
    }
    
    static void perm_dfs(int depth, int len, String number, boolean visit[]){
        if(depth == len){
            int num=Integer.valueOf(number);
            if(isPrime(num)){
                diction.add(num);
            }
            return;
        }
        for(int i=0; i<str.length(); i++){
            if(!visit[i]){
                visit[i] = true;
                perm_dfs(depth+1, len, number+str.charAt(i), visit);
                visit[i] = false;
            }
        }
    }
    static boolean isPrime(int num){
        if(num<=1)
            return false;
        
        for(int i=2; i<=(int)Math.sqrt(num); i++){
            if(num%i==0){
                return false;
            }
        }
        return true;
    }
}