[programmers] 모음사전 - Java

모음사전

문제 설명

제한사항

• word의 길이는 1 이상 5 이하입니다.

• word는 알파벳 대문자 'A', 'E', 'I', 'O', 'U'로만 이루어져 있습니다.

입출력 예

입출력 예 설명

solution.java

import java.util.*;

class Solution {
    
    List<String> list;
    String[] arr = {"A", "E", "I", "O", "U"};
    
    public int solution(String word) {
        int answer = 0;
        list = new ArrayList<>();
        
        // 사전 생성
        for(int i = 0 ; i < 5; ++i)
            makeWord(arr[i], 1);
        
        // 사전에서 단어 탐색
        for(int i = 0; i < list.size(); i++){
            if(list.get(i).equals(word))
                return i + 1;
        }
        return answer;
    }
    
    // dfs를 이용한 사전 생성
    public void makeWord(String base, int depth){
        list.add(base);
        if(depth == 5){
            return;
        }
        for(int i = 0 ; i < 5; ++i)
            makeWord(base + arr[i], depth + 1);
    }
}

핵심 키워드

• DFS 방식으로 사전을 생성한 후, 해당 사전에서 단어가 출현하는 순서를 찾아 리턴한다.
• 사전에서 해당 단어의 순서를 찾기 위해서는 순서를 보장하는 자료구조를 사용해야 하므로, List를 사용했다.

결론!