目录
1. 问题描述
你有一个带有四个圆形拨轮的转盘锁。每个拨轮都有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
解释:无法旋转到目标数字且不被锁定。
提示:
1 <= deadends.length <= 500
deadends[i].length == 4
target.length == 4
target
不在deadends
之中target
和deadends[i]
仅由若干位数字组成
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/open-the-lock
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题分析
图搜索最短路径问题,第一感是广度优先搜索。
针对每个4位数,下一步有8种可能。即在图中每个节点有8个邻接节点。图中总共有10000个节点。
为了提高查询效率,将deadends转换为哈希表(Python set),并且兼做visited,随后访问过的节点也加入deadends,以避免重复访问。
附加思考:
其实这个问题,如果没有deadends的约束,其实就是一个求4维空间中两个节点之间曼哈顿距离的问题。比如说,从0000到2134,很显然第一个数字需要拨2次,第二个数字需要拨1次,第三个数字需要拨3次,第四个数字需要拨4次,总共10次。当然目标数中大于5的数字反向拨更快,加上这个修正考虑即可。
但是有了deadends的限制,不能这样简单的计算,这相当于在方格网上不能按照最近距离(曼哈顿距离)路线去往目的地,而必须绕开deadends所包含的那些节点。比如说示例1所示情况。如何将deadends的约束融合到曼哈顿距离计算中去呢?从这个角度思考,能不能找到更高效的解决方案?
3. 代码
from typing import List
from collections import deque
class Solution:
def openLock(self, deadends: List[str], target: str) -> int:
if target in deadends or '0000' in deadends:
return -1
deadset = set(deadends)
q = deque([('0000',0)])
while len(q)>0:
node,layer = q.popleft()
if node == target:
return layer
# print(node,layer)
for k in range(4):
for i in [1,-1]:
nxt = node[:k] + str((int(node[k])+i) % 10) + node[k+1:]
if nxt not in deadset:
q.append((nxt,layer+1))
deadset.add(nxt)
return -1
if __name__ == "__main__":
sln = Solution()
deadends = ["0201","0101","0102","1212","2002"]
target = "0202"
print(sln.openLock(deadends, target))
deadends = ["8888"]
target = "0009"
print(sln.openLock(deadends, target))
deadends = ["8887","8889","8878","8898","8788","8988","7888","9888"]
target = "8888"
print(sln.openLock(deadends, target))
执行用时:604 ms, 在所有 Python3 提交中击败了58.06%的用户
内存消耗:16.4 MB, 在所有 Python3 提交中击败了42.58%的用户
官解还给出了一个基于A*-search的解法,瞄了几眼,默默地离开了(书读的太少,等学习一下A*-search再来回头看看)