乐趣区

关于算法:4位转盘锁算法

你有一个带有四个圆形拨轮的转盘锁。每个拨轮都有 10 个数字:'0', '1', '2', '3', '4', '5', '6', '7', '8', '9'。每个拨轮能够自在旋转:例如把 '9' 变为  '0''0' 变为 '9'。每次旋转都只能旋转一个拨轮的一位数字。

锁的初始数字为 '0000',一个代表四个拨轮的数字的字符串。

列表 deadends 蕴含了一组死亡数字,一旦拨轮的数字和列表里的任何一个元素雷同,这个锁将会被永恒锁定,无奈再被旋转。

字符串 target 代表能够解锁的数字,你须要给出最小的旋转次数,如果无论如何不能解锁,返回 -1。
示例 1:

 输出:deadends = ["0201","0101","0102","1212","2002"], target = "0202"
输入:6
解释:可能的挪动序列为 "0000" -> "1000" -> "1100" -> "1200" -> "1201" -> "1202" -> "0202"。留神 "0000" -> "0001" -> "0002" -> "0102" -> "0202" 这样的序列是不能解锁的,因为当拨动到 "0102" 时这个锁就会被锁定。

示例 2:

 输出: deadends = ["8888"], target = "0009"
输入:1
解释:把最初一位反向旋转一次即可 "0000" -> "0009"。

示例 3:

 输出: deadends = ["8887","8889","8878","8898","8788","8988","7888","9888"], target = "8888"
输入:-1
解释:无奈旋转到指标数字且不被锁定。

示例 4:

 输出: deadends = ["0000"], target = "8888"
输入:-1

提醒:

  1. 死亡列表 deadends 的长度范畴为 [1, 500]
  2. 指标数字 target 不会在 deadends 之中。
  3. 每个 deadendstarget 中的字符串的数字会在 10,000 个可能的状况 '0000''9999' 中产生。

解答:

func openLock(_ deadends: [String], _ target: String) -> Int {func upOne(_ z: String, _ index: Int) -> String {var s = Array(z)
        if s[index] == "9" {s[index] = "0"
        } else {s[index] = Character(String(s[index].wholeNumberValue! + 1))
        }
        return String(s)
    }
    
    func downOne(_ z: String, _ index: Int) -> String {var s = Array(z)
        if s[index] == "0" {s[index] = "9"
        } else {s[index] = Character(String(s[index].wholeNumberValue! - 1))
        }
        return String(s)
    }
        
    func dfs(_ x: String) -> Int {
        var num = 0
        let deadendSet = Set<String>(deadends)
        var visitedSet = Set<String>()
        var queue: [String] = []
        queue.append("0000")
        while !queue.isEmpty {
            for _ in 0..<queue.count {let value = queue.removeFirst()
                if deadendSet.contains(value) {continue}
                if value == target {return num}
                
                for j in 0..<4 {let vertex1 = downOne(value, j)
                    if !visitedSet.contains(vertex1) {visitedSet.insert(vertex1)
                        queue.append(vertex1)
                    }
                    let vertex2 = upOne(value, j)
                    if !visitedSet.contains(vertex2) {visitedSet.insert(vertex2)
                        queue.append(vertex2)
                    }
                }
            }
            num += 1
        }
        return -1
    }
    
    return dfs("0000")
}
退出移动版