본문 바로가기

프로그래머스 고득점kit/DFS, BFS 탐색

[고득점 kit]_DFS/BFS_#3. 단어 변환

728x90

(문제 설명)

두 개의 단어 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 합니다.
입출력 예begintargetwordsreturn
"hit" "cog" ["hot", "dot", "dog", "lot", "log", "cog"] 4
"hit" "cog" ["hot", "dot", "dog", "lot", "log"] 0
입출력 예 설명

예제 #1
문제에 나온 예와 같습니다.

예제 #2
target인 "cog"는 words 안에 없기 때문에 변환할 수 없습니다.

 


(문제 해설)

 

0. 문제를 보면 제시된 words에 있는 단어로만 바꿀 수 있습니다

1. 바꿀때 한 단어만 바꿀 수 있습니다.

 

2. 다들 생각해보면 bfs/ dfs로 탐색하면서 문자가 하나 차이 날 때만 바꾸게 하면 되고

3. 모두 순회했는데도 목표에 도달하지 못하면 0을 출력하게 하면 되겠다 하면 되겠다 싶고

4. 순회하 완료되고 목표에 도착하면 해당 순회 카운트 중 최소를 리턴하면 답이 나오겠다 싶을 겁니다.

 

5. 이렇게 문제를 간략화한 후 문제를 풀어보려고 시도해봅시다.

 


[ 코드 ]

 

# 3. 단어 변환
# 문자 하나가 다름을 체크
answer = 50


def solution(begin, target, words):
    global answer
    visited = [False] * 50
    
    dfs(begin, target, words, 0, visited)
    
    # 3. 모두 순회했는데 목표에 도달 못했을 때
    if answer == 50:
        return 0

    return answer


# 1. 단어가 하나가 다른지 체크
def chkOnedif(a, b):
    difCnt = 0

    for i in range(len(a)):
        if a[i] != b[i]:
            difCnt += 1

    if difCnt == 1:
        return True
    else:
        return False


def dfs(begin, target, words, cnt, visited):
    global answer
    
    # answer보다 커버리면 순회할 필요도 없음
    if cnt >= answer:
        return

    if begin == target:
        answer = min(answer, cnt)
        return

    for i in range(len(words)):
        if chkOnedif(begin, words[i]) and visited[i] == False:
            visited[i] = True

            dfs(words[i], target, words, cnt + 1, visited)

            visited[i] = False

( 코드 설명 )

0. 문제를 보면 제시된 words에 있는 단어로만 바꿀 수 있습니다

1. 바꿀 때 한 단어만 바꿀 수 있습니다.

 

dfs탐색을 할 때 첫 번째 word [i]가 0, 1 조건을 만족하면 방문 체크를 해줍니다

visited[i] = True

재귀를 이용해 다쉬 dfs를 순회합니다. => 순회 횟수 추가 cnt + 1

바뀐 단어 최신화 words [i]

dfs(words[i], target, words, cnt + 1, visited)

이렇게 다시 dfs로 들어가면 for문은 처음은 words [i]의 첫 번째 words [0]을 순회하게 될 것입니다.

즉, 모든 경우의 수를 dfs를 통하여 순회하게끔 설계된다는 것이죠.... 하지만 재방문은 하지 않습니다

visited에 의해 재방문을 방지했거든요.

이렇게 순회를 할걸 다하면

나오면서 방문 표시했던걸 다시 방문하지 않았다고 바꿔 줍니다.

visited[i] = False

이렇게 dfs재귀로 만들어진 모든 순회 구간이 다 순회되면 종료 조건에 의해 답이 출력되는 원리입니다.

 


후기

 

dfs를 문제에 맞게 구현하는 건 여간 까다롭습니다.

하지만 하다 보면 속도가 붙는 느낌이 드네요.

728x90