752. 打开转盘锁
752. 打开转盘锁
你有一个带有四个圆形拨轮的转盘锁。每个拨轮都有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
提示:
死亡列表 deadends 的长度范围为 [1, 500]。
目标数字 target 不会在 deadends 之中。
每个 deadends 和 target 中的字符串的数字会在 10,000 个可能的情况 '0000' 到 '9999' 中产生。
解析
这是个好问题,思路是用广度优先搜索来找到最短路径,有一些细节好好挖掘能形成编程技巧。 如果对BFS不熟,可以先参考这个题解:[778] 在水位上升的泳池中游泳
解法一:常规广度优先搜索
用一个集合存储当前访问状态列表
从 0000 开始搜索,对于每一个状态,可以扩展到最多 8 个状态,即将每一位增加 1 或减少 1,
将这些状态中没有搜索过并且不在 deadends 中的状态全部加入到队列中,并继续搜索,直到到达目标状态。
注意 0000 本身有可能也在 deadends 中。
为方便计数,可以在visited里维护到达每个状态的步数
存储当前访问状态列表的集合可以任选,切片、list、map等都可以,对顺序没有要求
func openLock(deadends []string, target string) int {
// 预处理,边界情况及时返回
const initial = "0000"
if target == initial {
return 0
}
isDead := make(map[string]bool, 0) // 将deadends处理成哈希表,方便迅速查找某个状态是否在里边
for _, v := range deadends {
isDead[v] = true
}
if isDead[initial] {
return -1
}
// 广度优先搜索
// BFS 准备
queue := list.New()
queue.PushBack(initial)
visited := map[string]int{initial: 0} // 已访问过的状态集合,并记录到达该状态花的步数
visitNextStatus := func(origin string) { // 获取origin的下一个状态(在这里有8种)并将合适的状态入队
b := []byte(origin)
for i, c := range b {
for d := -1; d <= 1; d += 2 { // 1 或 -1, b的每一位要加1或减1,以得到下一个状态
b[i] = byte((int(c-'0')+d+10)%10) + '0'
next := string(b)
b[i] = c // 恢复原字符串
if _, ok := visited[next]; ok || isDead[next] {
continue
}
visited[next] = visited[origin] + 1
queue.PushBack(next)
}
}
}
// BFS主体逻辑
for queue.Len() > 0 {
s := queue.Remove(queue.Front()).(string)
if s == target {
return visited[s]
}
visitNextStatus(s)
}
return -1
}
解法二:优化后的广度优先搜索
首先一开始可以把deadends里的状态放入visited,这样不用另开辟isDead的空间;实际测试这个优化对性能提升并不明显
看了leetcode题解,有这样一个优化:
可以分别从初始状态和目标状态向中间状态搜索,如果中间有状态相同,即完成了BFS
实际测试发现,维持这两个搜索使用的集合大小相当,会显著优化时间花费
注意,为了能迅速获知两个集合是不是有相同元素,这里的集合用哈希表再合适不过
func openLock(deadends []string, target string) int {
const initial = "0000"
if target == initial {
return 0
}
visited := make(map[string]bool)
for _, v := range deadends {
visited[v] = true
}
if visited[initial] {
return -1
}
start := map[string]bool{initial: true}
end := map[string]bool{target: true}
return bfs(start, end, visited, 0) // count代表步数,从0开始
}
// 模拟双向搜索。 count为步数
func bfs(start, end, visited map[string]bool, count int) int {
if len(start) > len(end) { // 从已经遍历少的那一端开始, 维持两端搜索的数量相当,能明显优化搜索步数
return bfs(end, start, visited, count)
}
if len(start) <= 0 {
return -1
}
nextStatus := make(map[string]bool) //存储start端下一步需要处理的状态
for s := range start {
if _, ok := end[s]; ok { // end队列也有,说明从初始到目标状态的一个通路形成了
return count
}
visited[s] = true
b := []byte(s)
for i, c := range b {
for d := -1; d <= 1; d += 2 { // 1 或 -1, b的每一位要加1或减1,以得到下一个状态
b[i] = byte((int(c-'0')+d+10)%10) + '0'
next := string(b)
b[i] = c // 复原状态
if visited[next] {
continue
}
nextStatus[next] = true
}
}
}
count++
return bfs(nextStatus, end, visited, count)
}