https://school.programmers.co.kr/learn/courses/30/lessons/43163
문제 설명
두 개의 단어 begin, target과 단어의 집합 words가 있습니다. 아래와 같은 규칙을 이용하여 begin에서 target으로 변환하는 가장 짧은 변환 과정을 찾으려고 합니다.
- 한 번에 한 개의 알파벳만 바꿀 수 있습니다.
- 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 합니다.
문제 해결 방법
탐색 문제이다.
words에서 특정 문자열의 index를 i라고 할 때, 다음과 같은 정보가 필요하다.
- dp[i] = 처음 문자열(begin) 부터 words[i] 문자열까지 변환 과정을 거친 횟수
코드 구현
def solution(begin, target, words):
answer = 999
words.append(begin)
dp = [999] * len(words)
length = len(begin)
def dfs(n, it):
dp[n] = min(dp[n], it)
for m in range(len(words)):
if dp[m] < it:
continue
cw = 0
ww = 0
for i in range(length):
if words[n][i] == words[m][i]:
cw += 1
else:
ww += 1
if ww == 1:
dfs(m, it+1)
dfs(len(words)-1, 0)
if target not in words:
return 0
answer = dp[words.index(target)]
if answer == 999:
return 0
return answer의사코드를 따로 적어볼까 했지만 역시 파이썬답게 의사코드랑 되게 유사해서 그냥 생략했다.
현재 탐색하려는 문자열 인덱스를 n이라고 할 때, 모든 문자열 words[m] 에 대해서 다음을 확인한다.
- 변환 과정이 이전보다 길어지진 않았는지
- 그 문자열로 변환할 수 있는지, 즉 알파벳 딱 하나 차이인지.
두 조건을 만족하면 그 문자열을 다음 탐색 대상으로 선정할 수 있으므로 dfs를 수행할 수 있다.
마지막엔 dp[target의 인덱스] 를 보면 target 문자열까지의 변환 과정 횟수를 구할 수 있다.
느낀점
문제를 풀 때에도 좀 체계적으로 설계하는 습관을 들여볼까
