9-5 코딩테스트 연습 - 단어 변환

문제 설명

두 개의 단어 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 합니다.

입출력 예

코드

import Foundation

func solution(_ begin: String, _ target: String, _ words: [String]) -> Int {
    // words에 target이 없으면 변환 불가
    if !words.contains(target) {
        return 0
    }
    
    // 두 단어가 한 글자 차이인지 확인하는 함수
    func isOneLetterDifferent(_ word1: String, _ word2: String) -> Bool {
        let arr1 = Array(word1)
        let arr2 = Array(word2)
        var differenceCount = 0
        
        for i in 0..<arr1.count {
            if arr1[i] != arr2[i] {
                differenceCount += 1
                if differenceCount > 1 {
                    return false
                }
            }
        }
        return differenceCount == 1
    }
    
    // BFS 탐색
    var queue: [(word: String, steps: Int)] = [(begin, 0)]
    var visited: Set<String> = []
    
    while !queue.isEmpty {
        let current = queue.removeFirst()
        let currentWord = current.word
        let currentSteps = current.steps
        
        // 현재 단어가 target이면 결과 반환
        if currentWord == target {
            return currentSteps
        }
        
        // 방문하지 않은 단어 중 한 글자 차이 단어를 큐에 추가
        for word in words {
            if !visited.contains(word) && isOneLetterDifferent(currentWord, word) {
                visited.insert(word)
                queue.append((word, currentSteps + 1))
            }
        }
    }
    
    // 변환할 수 없는 경우 0 반환
    return 0
}