752. 打开转盘锁
你有一个带有四个圆形拨轮的转盘锁。每个拨轮都有10个数字: '0', '1', '2', '3', '4', '5', '6', '7', '8', '9' 。每个拨轮可以自由旋转:例如把 '9' 变为 '0','0' 变为 '9' 。每次旋转都只能旋转一个拨轮的一位数字。
锁的初始数字为 '0000' ,一个代表四个拨轮的数字的字符串。
列表 deadends 包含了一组死亡数字,一旦拨轮的数字和列表里的任何一个元素相同,这个锁将会被永久锁定,无法再被旋转。
字符串 target 代表可以解锁的数字,你需要给出最小的旋转次数,如果无论如何不能解锁,返回 -1。
死亡列表 deadends 的长度范围为 [1, 500]。目标数字 target 不会在 deadends 之中。
每个 deadends 和 target 中的字符串的数字会在 '0000' 到 '9999' 中产生。
from queue import Queue
class Solution:
def openLock(self, deadends: List[str], target: str) -> int:
deadends = set(deadends) # in 操作在set中时间复杂度为O(1)
if '0000' in deadends:
return -1
if target == '0000':
return 0
# -------------------------------BFS 开始----------------------------
# 初始化根节点
q = Queue()
q.put(('0000', 0)) # (当前节点值,转动步数)
# 开始循环队列
while not q.empty():
# 取出一个节点
node, step = q.get()
# 放入周围节点
for i in range(4):
for add in (1, -1):
cur = node[:i] + str((int(node[i]) + add) % 10) \
+ node[i + 1:]
if cur == target:
return step + 1
if cur not in deadends:
q.put((cur, step + 1))
deadends.add(cur) # 避免重复搜索
# ------------------------------------------------------------------
return -1