두 개의 단어 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 합니다.
def solution(begin, target, words):
global answer
answer = 51
if target not in words: return 0
#words의 단어 중 알파벳이 한개의 차이가 있는 지 확인.
def find(str1, str2):
check = 0
for s1, s2 in zip(str1, str2):
if s1 != s2:
check += 1
return check == 1
def dfs(words, ss, cnt):
global answer
if ss == target:
answer = min(cnt, answer)
return
for i in range(len(words)):
if find(ss, words[i]):
dfs(words[:i] + words[i+1:],words[i],cnt+1)
dfs(words,begin,0);
return answer
dfs로 리스트를 넘겨줄 때
words[:i] + words[i+1:] 이렇게 한 이유는
해당 words[i]는 한번 사용했기 때문에 제외하고 새로운 리스트로 만들어서 파라미터로 넘겨줍니다.
리스트 크기가 하나씩 줄면 점점 탐색하는 속도가 빨라져서 시간초과가 나지 않습니다.
from collections import deque
def solution(begin, target, words):
#words의 단어 중 알파벳이 한개의 차이가 있는 지 확인.
def find(str1, str2):
check = 0
for s1, s2 in zip(str1, str2):
if s1 != s2:
check += 1
return check == 1
Q = deque()
Q.append([begin,[]]) #처음엔 비긴과 빈리스트를 넣음
while Q:
cur, save = Q.popleft()
if cur == target:
return len(save) #사용한 단어 수 리턴
for word in words:
if find(word,cur) and word not in save:
tmp = save[:] #깊은 복사
tmp.append(word)
Q.append([word,tmp])
return 0
save는 사용한 단어들을 저장하는 리스트입니다.
보통 DFS에서 checked나 visited 리스트를 사용하는 것과 비슷합니다.
원래는 permutations함수를 사용해서 모든 순열을 구하려고 했지만,
조건에서 words의 최대 갯수가 50개라 시간초과가 발생합니다.