https://programmers.co.kr/learn/courses/30/lessons/42839
import java.util.*;
class Solution {
public int solution(String numbers) {
int answer = 0;
HashSet<Integer> set = new HashSet<>();
permutation("", numbers, set);
Iterator<Integer> iter = set.iterator();
while (iter.hasNext()) {
int next = iter.next();
if (next == 2 || (next % 2 != 0 && isPrime(next))) {
answer++;
}
}
return answer;
}
public void permutation(String prefix, String str, HashSet<Integer> set) {
if (!prefix.equals("")) {
set.add(Integer.parseInt(prefix));
}
if (str.equals("")) {
return;
}
for (int i = 0; i < str.length(); i++) {
permutation(prefix + str.charAt(i), str.substring(0, i) + str.substring(i + 1), set);
}
}
public boolean isPrime(int x) {
if (x == 0 || x == 1) {
return false;
}
for (int i = 3; i <= (int)Math.sqrt(x); i += 2) {
if (x % i == 0) {
return false;
}
}
return true;
}
}
파이썬에서는 permutation
라이브러리를 이용해서 쉽게 풀었지만 자바에선 불가능하다. 재귀를 통해 각각 number
로 만들수있는 수를 Set
에 넣고 isPrime
함수를 통해 소수를 판별후 answer
개수를 리턴한다.