[프로그래머스] Lv.3 단어변환 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 함수를 작성해주세요.

 

 

DFS

def solution(begin, target, words):
    if target not in words:
        return 0
    results = []
    visited = [False] * len(words)
    dfs(begin, target, words, 0, results, visited)
    return min(results)

#다른 문자 개수 체크
def diff_check(a, b):
    cnt = 0
    for i in range(len(a)):
        if a[i] != b[i]:
            cnt += 1
    if cnt == 1:
        return True
    else:
        return False


def dfs(pre, target, words, change_count, results, visited):
    
    if pre == target:
        results.append(change_count)
        return
    
    for i in range(len(words)):
        if not visited[i]:
            if diff_check(pre, words[i]):
                visited[i] = True
                dfs(words[i], target, words, change_count+1, results, visited)
                visited[i] = False

 

처음에는 diff_check함수를 따로 작성하지 않아서 재귀 깊이 초과 오류가 발생했다. 따로 작성하지 않으면 스택에 쌓여 함수 호출이 필요한 메모리 공간이 스택에 계속 쌓여서 오류가 발생한다!

 

BFS

from collections import deque

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

def diff_check(a, b):
    cnt = 0
    for i in range(len(a)):
        if a[i] != b[i]:
            cnt += 1
    if cnt == 1:
        return True
    else:
        return False

def bfs(begin, target, words):
    queue = deque()
    queue.append([begin, 0])
    while queue:
        pre, depth = queue.popleft()
        if pre == target:
            return depth
        for word in words:
            cnt = 0
            if diff_check(pre, word):
                queue.append([word,depth+1])

BFS를 통해 풀었다. queue에 바꿀 단어와 그 depth를 배열로 저장하였다.

바꿀 단어와 words 배열안의 단어가 1자리만 다르다면 큐에 넣어 반복을 수행했다.