문제
문제 설명
두 개의 단어 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 합니다.
입출력 예
| 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 안에 없기 때문에 변환할 수 없습니다.
설계
solution
def solution(begin, target, words):
return bfs(begin,target,words)이 문제는 BFS로 풀 수 있다.
각 단어들을 노드로 삼고 시작 단어로부터 목표 단어까지의 최단 거리를 구하면 된다.
bfs
def bfs(begin,target,words):
q = deque([(begin,0)])
visited = make_visited(words)
visited[begin] = True
while q:
cur,cnt = q.popleft()
if cur == target:
return cnt
for w in words:
if not visited[w] and check(cur,w):
q.append((w,cnt+1))
visited[w] = True
return 0 기본적인 BFS 구조
make_visited
def make_visited(words):
visited = dict()
for w in words:
visited[w] = False
return visited단어의 방문 여부를 딕셔너리로 표현해주는 함수
check
def check(word0,word1):
cnt = 0
for i in range(len(word0)):
if word0[i] != word1[i]:
cnt += 1
if cnt >= 2:
return False
return True두 단어가 문제 조건에 부합하는지 검사해주는 함수
두 단어를 비교했을 때, 다른 글자가 2개 이상이라면 False
1개 이하라면 True를 return
전체 코드
from collections import deque
def make_visited(words):
visited = dict()
for w in words:
visited[w] = False
return visited
def check(word0,word1):
cnt = 0
for i in range(len(word0)):
if word0[i] != word1[i]:
cnt += 1
if cnt >= 2:
return False
return True
def bfs(begin,target,words):
q = deque([(begin,0)])
visited = make_visited(words)
visited[begin] = True
while q:
cur,cnt = q.popleft()
if cur == target:
return cnt
for w in words:
if not visited[w] and check(cur,w):
q.append((w,cnt+1))
visited[w] = True
return 0
def solution(begin, target, words):
return bfs(begin,target,words)
0x68656C6C6F21