[프로그래머스 - lv3] 단어 변환 - python

문제링크: https://programmers.co.kr/learn/courses/30/lessons/43163

코딩테스트 연습 - 단어 변환

---

문제 설명

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

입출력 예

 

 

 

문제접근

BFS를 사용해 풀었습니다.

자세한 풀이는 소스코드에 기재했습니다.

 

 

소스코드

"""
    BFS 알고리즘을 사용해 풀었습니다.
    먼저 words 배열에 target이 들어 있지 않으면 정답을 구할 수 없으므로 0을 반환합니다.
    begin 문자열이 만약 words에 들어 있지 않으면 words에 begin을 삽입하고
    bfs를 begin부터 시작합니다.
    words에 begin이 들어 있는지 확인하는 이유는, begin이 words에 들어 있을 경우
    bfs 중간에 begin 문자열을 또 탐색해 꼬일 수 있기 때문입니다.
    따라서 방문처리를 위해 words에 begin이 없을 경우 삽입하고 탐색이 시작될떄 방문처리를 해주어서
    중복 탐색을 방지합니다. 
"""
 
from collections import deque
 
 
def is_change(now, x):
    """ 문자열 now와 x가 한글자만 차이나는지 확인. """
    
    length = len(now)
    cnt = 0
    for i in range(length):
        if now[i] != x[i]:
            cnt += 1
        if cnt > 1:
            return False
 
    if cnt == 1:
        return True
    else:
        return False
 
 
def bfs(begin, target, words):
    """ 넓이 우선 탐색으로 begin에서 target까지의 최소 거리를 구함. """
    
    length = len(words)
    visited = [False] * length
 
    queue = deque()
    queue.append((begin, 0, len(words) - 1))
 
    while queue:
        now, depth, idx = queue.pop()
 
        if now == target:
            return depth
 
        if depth == length:
            continue
 
        visited[idx] = True
 
        for i, x in enumerate(words):
            if not visited[i] and is_change(now, x):
                queue.append((x, depth + 1, i))
 
    return 0
 
 
def solution(begin, target, words):
    
    if target not in words:
        return 0
 
    if begin not in words:
        words.append(begin)
    
    return bfs(begin, target, words)
 
 
def main():
    begin = "hit"
    target = "cog"
    words = ["hot", "dot", "dog", "lot", "log", "cog"]
    print(solution(begin, target, words))
    words = ["hot", "dot", "dog", "lot", "log"]
    print(solution(begin, target, words))
 
 
if __name__ == '__main__':
    main()