DFS/BFS - 단어 변환

• 문제 설명
  두 개의 단어 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 안에 없기 때문에 변환할 수 없습니다.

풀이

스택을 이용한 DFS로 풀이하였다.

1. words에 target이 존재하는지 확인하고, 존재하지 않으면 0을 리턴한다.

2. stack을 생성하여 begin을 넣는다.

3. words에서 begin과 철자가 하나만 다른 단어들을 찾아 next에 넣는다.

4. 탐색을 마치면 answer을 1만큼 증가시킨다.

5. next에 target이 있으면 answer을 리턴하고, 그렇지 않으면 stack을 next로 바꿔 탐색을 실행한다.

def solution(begin, target, words):
    answer = 0
    stack = [begin]
    
    if target not in words:
        return 0
    
    while len(words) != 0:
        next = []
        for item in stack:
            for word in words:
                count = 0
                for i in range(len(item)):
                    if item[i] != word[i]:
                        count += 1
                    if count == 2:
                        break
                if count == 1:
                    next.append(word)
                    words.remove(word)
        answer += 1
        if target in next:
            return answer
        else:
            stack = next
            
    return answer