[프로그래머스] 단어 변환 / Swift

[문제 보기]

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

입출력 예

begin, target, words, return

"hit"	"cog"	["hot", "dot", "dog", "lot", "log", "cog"]	4
"hit"	"cog"	["hot", "dot", "dog", "lot", "log"]	0

입출력 예 설명

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

 

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

 

https://programmers.co.kr/learn/courses/30/lessons/43163

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

 

---

words 배열을 토대로 그래프를 구성하고 begin에 해당하는 노드부터 target에 해당하는 노드까지 최단거리를 구하는 문제였다.

문제 조건에서 한번에 최대 1개의 알파벳만 바꿀 수 있는 규칙을 모든 간선 가중치가 1인 그래프로 이해했다.

그리고 모든 간선값이 1인 다익스트라로 풀었다.

 

import Foundation

func getDistance(_ s1: String, _ s2: String) -> Int {
    return zip(Array(s1), Array(s2)).filter { $0 != $1 }.count
}

func solution(_ begin:String, _ target:String, _ words:[String]) -> Int {
    let words = [begin] + words
    guard let targetIndex = words.firstIndex(of: target) else { return 0 }
    let n = words.count
    var graph = [[Int]](repeating: [Int](repeating: 0, count: n+1), count: n+1)
    for i in (1..<n) {
        for j in (i+1...n) {
            if getDistance(words[i-1], words[j-1]) == 1 {
                graph[i][j] = 1
                graph[j][i] = 1
            }
        }
    }
    
    func dijkstra(_ start: Int, _ to: Int) -> Int {
        let INF = 987654321
        var distance = [Int](repeating: INF, count: n+1)
        var queue = [start]
        distance[start] = 0
        while !queue.isEmpty {
            let first = queue.removeFirst()
            for i in (1...n) where graph[first][i] != 0 {
                let cost = distance[first] + 1 // 1로 고정
                if distance[i] > cost {
                    distance[i] = cost
                    queue.append(i)
                }
            }
        }
        return distance[to]
    }

    return dijkstra(1, targetIndex + 1)
}

 

---

생각 정리

다익스트라에서 우선순위 큐 vs 큐

 

https://www.acmicpc.net/board/view/26887

글 읽기 - 큐 vs 우선순위_큐 다익스트라 구현