문제 설명
한자리 숫자가 적힌 종이 조각이 흩어져있습니다. 흩어진 종이 조각을 붙여 소수를 몇 개 만들 수 있는지 알아내려 합니다.
각 종이 조각에 적힌 숫자가 적힌 문자열 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은 같은 숫자로 취급합니다.
https://programmers.co.kr/learn/courses/30/lessons/42839
코딩테스트 연습 - 소수 찾기
한자리 숫자가 적힌 종이 조각이 흩어져있습니다. 흩어진 종이 조각을 붙여 소수를 몇 개 만들 수 있는지 알아내려 합니다. 각 종이 조각에 적힌 숫자가 적힌 문자열 numbers가 주어졌을 때, 종이
programmers.co.kr
내 소스
import java.util.*;
class Solution {
int[] arr;
int[] output;
boolean[] visited;
int n;
int answer;
Set<String> hashSet;
public int solution(String numbers) {
n = numbers.length();
arr = new int[n];
output = new int[n];
visited = new boolean[n];
hashSet = new HashSet<String>();
for(int i=0; i<n; i++){
arr[i] = numbers.charAt(i)-'0';
}
perm(0);
return answer;
}
boolean isPrime(int num){
if(num < 2){
return false;
}else if(num == 2){
return true;
}else{
for(int i=2; i<num; i++){
if(num%i==0){
return false;
}
}
return true;
}
}
void perm(int depth){
if(depth != 0){
String s = "";
for(int i=0; i<depth; i++){
s += output[i];
}
if(isPrime(Integer.parseInt(s))){
if(!hashSet.contains(s)){
answer++;
hashSet.add(s);
}
}
}
for(int i=0; i<n; i++){
//0으로 시작하는 숫자는 체크하지 않음
if(depth==1 && output[0]==0){
continue;
}
if(!visited[i]){
visited[i] = true;
output[depth] = arr[i];
perm(depth+1);
visited[i] = false;
}
}
}
}
* 사용한 클래스 : HashSet - contains(), add()
* 사용한 알고리즘 : 소수 판별, 순열
'Algorithm' 카테고리의 다른 글
[프로그래머스/정렬] 가장 큰 수 (0) | 2021.08.12 |
---|---|
[프로그래머스/정렬] K번째수 (0) | 2021.08.12 |
[프로그래머스/힙] 더 맵게 (0) | 2021.08.10 |
[프로그래머스/큐] 기능개발 (0) | 2021.08.10 |
[프로그래머스/그리디] 구명보트 (0) | 2021.08.08 |