단어변환 (python)

BFS with count 기본문제

프로그래머스 문제 단어변환 - LV3

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 함수를 작성해주세요.

입출력 예

begin	target	words				return
hit	cog	[hot, dot, dog, lot, log, cog]	4
hit	cog	[hot, dot, dog, lot, log]	0

솔루션

# cur_word, que
hit deque([('hot', 1)])
hot deque([('dot', 2), ('lot', 2)])
dot deque([('lot', 2), ('dog', 3)])
lot deque([('dog', 3), ('log', 3)])
dog deque([('log', 3), ('cog', 4)])
log deque([('cog', 4)])

코드

# 파이썬
from collections import deque

def can_change(cur_word, words):
    cand = []
    for word in words:
        diff = [True for x, y in zip(cur_word, word) if x != y]
        if len(diff) == 1: 
            cand.append(word)
    return cand

def solution(begin, target, words):
    visited = set([begin])
    que = deque([(begin, 0)])
    while que:
        cur_word, cur_count = que.popleft()
        # 확인
        if cur_word == target:
            return cur_count
        # 탐색
        for word in can_change(cur_word, words):
            if word not in visited:
                que.append((word, cur_count + 1))
                visited.add(word)
    return 0