코딩 테스트
(Java) 프로그래머스 - 모음사전
로승리
2022. 7. 22. 20:31
문제는 L2에 맞지 않게 너무 쉬웠다.
재귀를 사용할수만 있다면 바로 풀리는 문제다.
원래는 str이 word와 같다면 메서드를 바로 종료하고 싶어서 밑과 같이 작성했지만
모든 사이클을 무조껀 돌아야 했다.
if(word.equals(str)) {
answer = cnt;
return;
}
if(depth == 5) {
return;
}
그래서 그냥 모든 조합을 list에 넣고 선형 탐색으로 답을 구했다.
모든 조합을 다 해봐야 3600개쯤 밖에 안되니 시간 초과가 나진 않을 것 같았다.
최종 코드
import java.util.ArrayList;
import java.util.List;
class Solution {
static String[] arr;
static List<String> list;
public int solution(String word) {
int answer = 0;
list = new ArrayList<>();
arr = new String[]{"A", "E", "I", "O", "U"};
// 완전탐색 메서드 호출
recursion(word, "", 0);
// 선형 탐색
for (int i = 0; i < list.size(); i++) {
if(list.get(i).equals(word)) {
answer = i;
break;
}
}
return answer;
}
// 완전탐색 메서드
static void recursion(String word, String str, int depth) {
list.add(str);
if(depth == 5) {
return;
}
for (int i = 0; i < arr.length; i++) {
recursion(word, str + arr[i], depth + 1);
}
}
}