두 개의 단어 begin, target과 단어의 집합 words가 있습니다. 아래와 같은 규칙을 이용하여 begin에서 target으로 변환하는 가장 짧은 변환 과정을 찾으려고 합니다.
두 개의 단어 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 |
예제 #1
문제에 나온 예와 같습니다.
예제 #2
target인 cog는 words 안에 없기 때문에 변환할 수 없습니다.
from collections import deque
def compare(compare_word,words):
a=list()
for word in words:
if sum((1 if a!=b else 0) for a,b in zip(word,compare_word))==1:
a.append(word)
return a
def solution(begin, target, words):
ch=set([begin])
dq=deque([(begin,0)])
if target not in words: return 0
while dq:
start,cnt=dq.popleft()
if start==target: return cnt
for word in compare(start,words):
if word not in ch:
ch.add(word)
dq.append([word,cnt+1])
return 0
문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/43163