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

문제

---

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

https://school.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 합니다.

입출력 예begintargetwordsreturn

"hit"	"cog"	["hot", "dot", "dog", "lot", "log", "cog"]	4
"hit"	"cog"	["hot", "dot", "dog", "lot", "log"]	0

• 시작 문자열로 부터 문자를 한개씩만 바꾸어서 Target 문자열과 동일한 문자열을 만들 수 있는 최솟값.
• Words에 있는 단어로만 바꿀 수 있다.

 

알고리즘

---

• Idea = 매번 각 단어마다 단어를 바꾸어 BFS로 탐색

 

1. Deque를 만들어준다. ( 단어 , 바꾼 갯수 )

2. 현재 단어가 Target과 동일하다면 변경

3. 현재 단어에서 갈 수 있는 모든 경우의 수를 보고, Word_dict안에 있으면서 더 빠르게 도착 할 수 있는 값에 대해서만 처리한다.

 

코드

---

import sys
from collections import deque

def solution(begin, target, words):
    alphas = [chr(i) for i in range(97,97+26)]
    answer = 0
    word_dict = dict()
    n = len(begin)
    #visited 역할
    for word in words:
        word_dict[word] = sys.maxsize
        
    q = deque()
    q.append((begin,0))
    
    while q :
        word,cnt = q.popleft()
        # 종료 조건
        if word == target:
            answer=cnt
            break
        
        # 단어 변환
        for i in range(n):
            for alpha in alphas :
                temp = word[:i] + alpha + word[i+1:]
                if temp in word_dict and cnt+1 < word_dict[temp] :
                    word_dict[temp]=cnt+1
                    q.append( (temp,cnt+1))

    return answer

begin = "hit"
target = "cog"
words = ["hot", "dot", "dog", "lot", "log"]	

solution(begin, target, words)

 

코드 리뷰

---

• 중요 포인트 = Word에 있는 단어와 매칭되면서 매번 하나의 단어만 바꾸면서 Count한다.
• Generator를 이용하여 현재 단어와 Word가 갯수가 하나만 다른 경우만 찾아서 갱신 시켜주면 된다.
• 모든 경우의 수를 전부 찾지만, 찾고자 하는 단어만 탐색하기 때문에 빠르게 처리할 수 있으며 코드가 간결해진다.

 

1. generator를 만들어 현재 단어와 words에서 한가지만 다른 단어를 가진 단어를 추출

2. begin 단어부터 시작하여 새로운 단어가 갱신 될 때 마다 추가

from collections import deque

# 문자 목록에서 현재 단어와 한개의 문자만 다른 경우
def check_with_generator( current , words ) :
    for word in words :
        count = 0
        for c,w in zip(current,word) :
            if c!=w :
                count+=1

        if count==1 :
            yield word
            
def solution(begin, target, words):
    make_word = {begin:0}
    q = deque([begin])

    while q :
        current = q.popleft()
        for word in check_with_generator(current,words):
            # 방문 한적이 없다면 추가 visited 역할
            if word not in make_word :
                make_word[word] = make_word[current]+1
                q.append(word)
                
    return make_word.get(target , 0 )