[Programmers] 깊이/너비 우선 탐색(DFS/BFS) > 단어 변환 with python

문제 설명

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

 

입출력 예시

 

풀이과정

우선 한 번에 한 개의 알파벳만 바꿀 수 있다는 조건을 생각하며 현재의 word와 words 리스트 안의 word와 어떻게 비교하면 좋을지 생각했다.

그 때 나온 답이 zip()이다.

zip 함수로 리스트들을 짝지어주며 현재 word와 words 리스트의 word를 비교하는 함수 check()를 만들었다.

그리고 너비우선탐색을 진행하였다.

 

bfs라는 큐에 word와 그 때의 count를 튜플로 넣었고 words 리스트 안의 원소를 방문했는지 알아보기 위해 visit 리스트를 만들었다.

반복문을 돌리면서 조건에 해당되면 큐에 원소를 삽입하고 target이 나왔을 경우 그 때의 count를 return 해주었다.

 

코드

def solution(begin, target, words):
    bfs = [(begin, 0)] #queue
    visited = [begin]
    if target not in words:
        return 0

    while bfs:
        cur_word, cur_cnt = bfs.pop(0)
        if cur_word == target:
            return cur_cnt
        for word in words:
            if word not in visited:
                if check(cur_word, word)==True:
                    bfs.append((word, cur_cnt+1))
                    visited.append(word)

    return 0

def check(cur_word, word):
    cnt=0
    for a, b in zip(cur_word, word):
        if a==b:
            cnt+=1
    if cnt==len(word)-1:
        return True
    else:
        return False