Programmers_DFS|BFS_단어 변환

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

Link: https://programmers.co.kr/learn/courses/30/lessons/43163?language=python3

문제 설명

두 개의 단어 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

Code 접근

1. recursive 함수를 통해 변환이 가능할때 까지 계속 변환하고, 중복해서 변환은 안되도록 한다.
2. 변환을 계속하면서 target과 같은 값이면 그때 변환한 갯수를 구해주면 된다. 갯수들 중 가장 작은 값을 return 하도록 한다.

Code | Python

count = 0
answer = []
def DFS(start, target, words, used):
    global count
    global answer
    if start == target:
        answer.append(count)
    else:
        for cand in words:
            diff = 0
            for i in range(len(cand)):
                if start[i]!= cand[i]:
                    diff += 1
                if diff >= 2: break
            if diff == 1 and cand not in used:
                count +=1 
                used.append(cand)
                DFS(cand,target,words, used)
                used.pop()
                count -=1

def solution(begin, target, words):
    global answer
    if target not in words:
        return 0
    
    used = []
    DFS(begin, target, words, used)
    return min(answer)

Code Screen Shot & Output

#1
Input: print(solution("hit","cog",["hot", "dot", "dog", "lot", "log", "cog"]))
Output: 4