[Python/알고리즘] 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 합니다.

해결방법

1. 모든 경로를 탐색하고 그 중 최소 횟수를 반환해야한다. 따라서 BFS로 모든 노드를 동시 탐색한다.

2. BFS이므로 queue를 사용했고 이미 탐색한 word를 다시 탐색하지 않기 위해 queue에 현재 값과 지나온 경로를 담아준다.

3. 변환 가능한 단어를 검사하는 함수를 따로 작성한다. 글자 하나만 다른 경우에 True를 반환해준다.

4. BFS이므로 노드를 탐색하며 원하는 결과값이 나온 경우 즉시 반환한다.

 

 

해결 코드

from collections import deque
# queue를 위함


def check(s, begin):
    answer = 0
    for i in range(len(s)):
        if list(s)[i] != list(begin)[i]:
            answer += 1
    return True if answer == 1 else False


def solution(begin, target, words):
    if target not in words:
        return 0

    queue = deque()
    queue.append([begin, []])

    while queue:
        n, l = queue.popleft()
        for word in words:
            if word not in l and check(word, n):
                if word == target:
                    return len(l) + 1
                temp = l[0:]
                temp.append(word)
                queue.append([word, temp])

    return 0

begin 값과 경로를 기본으로 queue에 담아주고,

words를 검사해 가능하다면 해당 노드를 queue에 담아준다.