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

김지원·2022년 10월 25일
0
post-custom-banner

📄 문제 설명

두 개의 단어 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가 매개변수로 주어질 때, 최소 몇 단계의 과정을 거쳐 begintarget으로 변환할 수 있는지 return 하도록 solution 함수를 작성해주세요.

제한사항

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

🔨 문제해결

  • 가장 짧은 !! BFS로 문제 해결 가능하다.
  • 먼저 words 리스트에서 각 단어별로 한 개의 알파벳만 다른 단어들로 이루어진 딕셔너리를 만들어준다.
  • 그리고 BFS로 문제 풀이!
from collections import deque
def solution(begin, target, words):
    answer = 0
    stack=deque()
    graph={w:[] for w in words}
    checked=set()
    # make graph
    for i in range(len(words)):
        cnt=0
        for k in range(len(words[i])):
            if words[i][k]!=begin[k]:
                cnt+=1
        if cnt==1:
            stack.append((words[i],1))
        for j in range(i+1,len(words)):
            cnt=0
            for k in range(len(words[i])):
                if words[i][k]!=words[j][k]:
                    cnt+=1
            if cnt==1:
                graph[words[i]].append(words[j])
                graph[words[j]].append(words[i])
    while stack:
        word,step=stack.popleft()
        if word==target:
            return step
        if word in checked:
            continue
        checked.add(word)
        for next_w in graph[word]:
            if next_w not in checked:
                stack.append((next_w,step+1))
        
    return 0

https://school.programmers.co.kr/learn/courses/30/lessons/43163

profile
Make your lives Extraordinary!
post-custom-banner

0개의 댓글