[프로그래머스]단어 변환/43163번/Python/파이썬/DFS/BFS

heeee·2021년 2월 10일
0

algorithm

목록 보기
70/123

💡문제

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

입출력

begintargetwordsreturn
hitcog[hot, dot, dog, lot, log, cog]4
hitcog[hot, dot, dog, lot, log]0

입출력 예 설명

예제 #1
문제에 나온 예와 같습니다.

예제 #2
target인 cog는 words 안에 없기 때문에 변환할 수 없습니다.



📖내가 작성한 code

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

0개의 댓글