[프로그래머스 Level 3] 단어 변환

문제

 

두 개의 단어 begin, target과 단어의 집합 words가 있습니다. 아래와 같은 규칙을 이용하여 begin에서 target으로 변환하는 가장 짧은 변환 과정을 찾으려고 합니다.

 

1. 한 번에 한 개의 알파벳만 바꿀 수 있습니다.
2. words에 있는 단어로만 변환할 수 있습니다.

 

예를 들어 begin이 "hit", target가 "cog", words가 ["hot","dot","dog","lot","log","cog"]라면 "hit" -> "hot" -> "dot" -> "dog" -> "cog"와 같이 4단계를 거쳐 변환할 수 있습니다.

 

두 개의 단어 begin, target과 단어의 집합 words가 매개변수로 주어질 때, 최소 몇 단계의 과정을 거쳐 begin을 target으로 변환할 수 있는지 return 하도록 solution 함수를 작성해주세요.

 

 

제한 사항

• 각 단어는 알파벳 소문자로만 이루어져 있습니다.
• 각 단어의 길이는 3 이상 10 이하이며 모든 단어의 길이는 같습니다.
• words에는 3개 이상 50개 이하의 단어가 있으며 중복되는 단어는 없습니다.
• begin과 target은 같지 않습니다.
• 변환할 수 없는 경우에는 0를 return 합니다.

 

---

풀이

 

어렵지 않은 DFS 문제였다. 조건에 따라 탐색해주면 무난하게 PASS 할 수 있다.

 

이 문제의 핵심은 재귀 조건을 잘 적용하는 것!

한 번에 한 개의 알파벳만 바꿀 수 있습니다.

 

위 조건을 염두해두고 코드를 보자!

 

private void translate(String[] words, String now, int depth) {
        if(depth >= N) return;
        if(now.equals(target)) {
            answer = Math.min(answer, depth);
            return;
        }

        for(int i=0; i<N; i++) {
            int count = 0;

            for(int j=0; j<now.length(); j++) {
                if(now.charAt(j) != words[i].charAt(j)) count++;
            }
            if(count == 1 && !visited[i]) {
                visited[i] = true;
                translate(words, words[i], depth + 1);
                visited[i] = false;
            }
        }
    }

 

평범한 DFS 메서드라고 보면 되는데 단지 현재 단어(now)와 다음 후보 단어를 비교해서 한 개의 알파벳만 다를 경우 재귀를 반복한다.

 

또한, target이 words 배열 안에 들어있지 않으면 탐색을 할 필요가 없으므로, target이 words 배열 안에 있는지 확인해주면 된다.

 

전체 코드

 

public class Solution {
    private boolean[] visited;
    private String target;
    private int answer = 0;
    private int N;

    private void translate(String[] words, String now, int depth) {
        if(depth >= N) return;
        if(now.equals(target)) {
            answer = Math.min(answer, depth);
            return;
        }

        for(int i=0; i<N; i++) {
            int count = 0;

            for(int j=0; j<now.length(); j++) {
                if(now.charAt(j) != words[i].charAt(j)) count++;
            }
            if(count == 1 && !visited[i]) {
                visited[i] = true;
                translate(words, words[i], depth + 1);
                visited[i] = false;
            }
        }
    }

    public int solution(String begin, String target, String[] words) {
        this.target = target;

        N = words.length;
        visited = new boolean[N];

        for(String word : words) {
            if(target.equals(word)) {
                answer = Integer.MAX_VALUE;
                break;
            }
        }
        if(answer != 0) {
            translate(words, begin, 0);
        }

        return answer;
    }
}