코딩 테스트

(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);
        }
    }
}