단어 변환

문제 설명

두 개의 단어 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 합니다.

 

입출력 예

begin	target	words	return
"hit"	"cog"	["hot", "dot", "dog", "lot", "log", "cog"]	4
"hit"	"cog"	["hot", "dot", "dog", "lot", "log"]	0

 

입출력 예 설명

예제 #1
문제에 나온 예와 같습니다.

예제 #2
target인 "cog"는 words 안에 없기 때문에 변환할 수 없습니다.

 

---

 

풀이 방법 생각

bfs(너비 우선 탐색) 방법으로 제일 빠른 경로를 찾아야 되는 문제라고 생각했다. Node class를 만들어 구현한 것이 아닌 약간 느낌만(?) 사용했다. 현재 노드와 다음 탐색할 노드를 담을 queue를 사용해서 현재 노드를 제거하면서 다음 경로를 뒤에 삽입해주는 방식으로 구현했다. 중간에 결과값이 나오면 반복문을 종료했고 words를 모두 한번씩 갔더라도 없으면 종료해줬다. equals 함수로 String을 비교할 수도 있는데 String이 몇 글자 다른지 판단하기 위해 함수를 이용해서 비교해줬다.

 

내 코드

import java.util.LinkedList;
import java.util.Queue;

class Solution {
    public int solution(String begin, String target, String[] words) {
        int answer = 0;
        boolean answerChk = false;

        Queue<String> q = new LinkedList<String>();

        q.offer(begin);

        while (!answerChk && answer <= words.length) {
            int size = q.size();

            for (int i = 0; i < size; i++) {
                String word = q.poll();
                if (wordsEq(word, target) == 0) {
                    answerChk = true;
                    break;
                }
                for (int j = 0; j < words.length; j++) {
                    if (wordsEq(word, words[j]) == 1) {
                        q.offer(words[j]);
                    }
                }
            }
            if (!answerChk)
                answer++;
        }

        if (answer > words.length)
            answer = 0;

        return answer;
    }

    int wordsEq(String a, String b) {
        int count = 0;
        for (int i = 0; i < a.length(); i++) {
            if (a.charAt(i) != b.charAt(i))
                count++;
        }
        return count;
    }
}

 

다른 사람 풀이

import java.util.LinkedList;
import java.util.Queue;

class Solution {

    static class Node {
        String next;
        int edge;

        public Node(String next, int edge) {
            this.next = next;
            this.edge = edge;
        }
    }

    public int solution(String begin, String target, String[] words) {
        int n = words.length, ans = 0;

        // for (int i=0; i<n; i++)
        //  if (words[i] != target && i == n-1) return 0;

        Queue<Node> q = new LinkedList<>();


        boolean[] visit = new boolean[n];
        q.add(new Node(begin, 0));

        while(!q.isEmpty()) {
            Node cur = q.poll();
            if (cur.next.equals(target)) {
                ans = cur.edge;
                break;
            }

            for (int i=0; i<n; i++) {
                if (!visit[i] && isNext(cur.next, words[i])) {
                    visit[i] = true;
                    q.add(new Node(words[i], cur.edge + 1));
                }
            }
        }

        return ans;
    }

    static boolean isNext(String cur, String n) {
        int cnt = 0;
        for (int i=0; i<n.length(); i++) {
            if (cur.charAt(i) != n.charAt(i)) {
                if (++ cnt > 1) return false;
            }
        }

        return true;
    }    
}

 

Node 클래스를 만들어서 다음 노드로 넘어가면서 bfs를 구현했다. 그래서 반복문의 시간 복잡도가 내가 풀이한 방법보다 적게 나온 것으로 보인다. 이 풀이로 채점한 결과 5개의 테스트 중 한 테스트의 시간이 내 코드보다 짧게 나왔다. 내 코드는 방문한지 판단하지 않아서 그런 거 같다. 테스트 케이스가 뭔지 모르겠지만 왠지 데이터가 커지면 방문했더라도 다시 방문하도록 되어있어서 내 코드의 효율성이 떨어지는 것으로 생각된다.