두 개의 단어 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 함수를 작성해주세요.
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 solution(begin, target, words):
answer = 0
queue=deque([[begin,0]])
visited=[]
while queue:
node,dis=queue.popleft()
if node==target:
answer=dis
break
for i in range(len(words)):
cnt=0
if words[i] not in visited:
for j in range(len(node)):
if node[j]!=words[i][j]:
cnt+=1
if cnt==1:
queue.append([words[i],dis+1])
visited.append(words[i])
return answer
BFS를 이용하여 푸는 문제였다. 현재 가리키는 값과 한 글자 차이나는 단어들을 찾고 한 번 더 BFS를 돌리는 방식
BFS를 돌리다가 target값을 찾으면 멈추고 이때까지의 거리를 answer에 담아 반환한다.
거리 계산
여러 번 풀었으나 결국 다른 풀이들을 보면서 공부한 문제. BFS를 사용하려고 했으나 문제에 적용하는 게 힘들었다. 익숙해져서 잘 풀 수 있다고 생각했는데 더 고민해보고 공부해봐야겠다는 생각이 들었다.
거리를 계산하는 방법을 늘 헷갈려했는데 이 참에 정리까지 했으니 잘 기억해야겠다.