[Python] 프로그래머스 단어 변환

프로그래머스 레벨 3 

문제 설명

두 개의 단어 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 / BFS 문제

단어가 변환되려면 1개의 알파벳만 달라야하고, 최소 단계를 찾아야함.

BFS 활용

시작 단어와 현재 단계 수를 큐에 추가

만약 변환 가능한 단어라면 큐에 추가하면서 단계 수 1 증가

만약 현재 단어가 최종 target과 일치한다면 현재 단계 수 반환

from collections import deque

def bfs(begin, target, words):
    q = deque()
    q.append((begin, 0))
    
    while q:
        current, step = q.popleft()
        if current == target:
            return step
        
        for word in words:
            cnt = 0
            for i in range(len(word)):
                if current[i] != word[i]:
                    cnt += 1
            if cnt == 1:
                q.append((word, step + 1))
            
def solution(begin, target, words):
    if target not in words:
        return 0
    else:
        return bfs(begin, target, words)

위의 코드도 정답처리는 되는데 방문한 단어를 반복하여 다시 방문할 필요 없으므로 

visited 사용해서 방문한 노드는 다시 방문하지 않도록 처리하면 시간 훨씬 단축

from collections import deque

def bfs(begin, target, words, visited):
    q = deque()
    q.append((begin, 0))
    
    while q:
        current, step = q.popleft()
        if current == target:
            return step
        
        for idx, word in enumerate(words):
            if not visited[idx]:
                cnt = 0
                for i in range(len(word)):
                    if current[i] != word[i]:
                        cnt += 1
                if cnt == 1:
                    q.append((word, step + 1))
                    visited[idx] = True
            
def solution(begin, target, words):
    visited = [False] * len(words)
    
    if target not in words:
        return 0
    else:
        return bfs(begin, target, words, visited)

---

https://school.programmers.co.kr/learn/courses/30/lessons/43163

프로그래머스