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

문제

두 개의 단어 begin, target과 단어의 집합 words가 있을 때, 아래와 같은 규칙을 이용하여 begin에서 target으로 변환하는 가장 짧은 변환 과정을 찾으려고 합니다.

• 한 번에 한 개의 알파벳만 바꿀 수 있습니다.
• words에 있는 단어로만 변환할 수 있습니다.

두개의 단어 begin, target과 단어의 집합 words가 매개변수로 주어질 때, 최소 몇 단계의 과정을 거쳐 begin을 target으로 변환할 수 있느지 return 하도록 solution 함수를 작성해주세요.

제한사항

• 각 단어는 알파벳 소문자로만 이루어져 있습니다.
• 각 단어의 길이는 3 이상 10 이하이며 모든 단어의 길이는 같습니다.
• words에는 3개 이상 50개 이하의 단어가 있으며 중복되는 단어는 없습니다.
• begin과 target은 같지 않습니다.
• 변환할 수 없는 경우에는 0을 return 합니다.

풀이

1. 각 단어들이 변환이 가능한 경우에는 서로 연결되어 있다고 가정합니다.

   ex.
   hit - hot
   hot - dot
   dot - dog

2. 너비우선탐색을 통해 begin에서 target까지 가는데 최소 몇 단계의 과정이 거쳐야 하는지 계산해줍니다.

코드

def solution(begin, target, words):
    
    if target not in words:
        return 0
    
    maps = {begin: []}
    for w in words:
        maps[w] = []
    answer, n = 0, len(maps)
    
    keys = list(maps.keys())
    for i in range(n):
        for j in range(i+1,n):
            if i != j and connect(keys[i],keys[j]):
                maps[keys[i]].append(keys[j])
                maps[keys[j]].append(keys[i])
    
    q = list()
    q.append((0, begin))
    visited = list()
    while q:
        cost, word = q.pop(0)
        
        if word == target:
            return cost
        
        for w in maps[word]:
            if w not in visited:
                visited.append(w)
                q.append((cost+1, w))
                
    return 0

def connect(a, b):
    diff = 0
    for i in range(len(a)):
        if a[i] != b[i]:
            diff += 1
    # 한 번의 한 개의 알파벳만 변경 가능
    return diff < 2