코딩테스트 - [Swift] 단어변환(BFS)

단어 변환

문제 설명

두 개의 단어 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 안에 없기 때문에 변환할 수 없습니다.

 

접근 방법

한 알파벳씩 변환할 수 있기 때문에 한 알파벳씩 변경할 수 있는 단어가 있는지 확인합니다.

이후 단어를 변경해가며 어느 레벨에서 target 단어가 나오는지 찾아갑니다.

해당 문제를 그래프로 그려보자면 아래처럼 나옵니다.

 

hit에서 변경 가능한 단어는 hot뿐입니다.

hit에서 변경 가능한 단어는 dot과 lot입니다.

이처럼 레벨을 내려가며 몇레벨에 있는지 확인하면 됩니다.

코드

import Cocoa

import Foundation

func solution(_ begin:String, _ target:String, _ words:[String]) -> Int {
    
    //한 가지 단어가 다른지 판단하는 함수
    func diffOne(_ word1: String, _ word2: String) -> Bool{
        var count = 0
    
        if(word1.count != word2.count){
            return false
        }
        
        let arrword1 = Array(word1)
        let arrword2 = Array(word2)
        
        for i in 0..<arrword1.count{
            if arrword1[i] != arrword2[i]{
                count += 1
            }
            
            //다른 단어의 개수가 1개 이상이면 false를 return합니다/
            if count > 1{
                return false
            }
        }
        
        return true
    }
    
    //target 단어가 words에 포함되어있지 않으면 0을 return 합니다.
    if !words.contains(target){
        return 0
    }
    
    //배열을 만들어 큐로 사용합니다.
    var queue = [String]()
    queue.append(begin)
    
    var visited = Array(repeating: false, count: words.count)
    var count = 0
    
    //큐가 비어있지 않은 동안 반복합니다.
    while !queue.isEmpty{
        let now = queue.popLast()!
        if(now == target) { break }
                
        for idx in 0..<words.count{
            if !visited[idx]{
                if diffOne(now, words[idx]){
                    queue.append(words[idx])
                    visited[idx] = true
                }
            }
        }
        count += 1
        
    }
    
    
    return count
}