leetcode 752. 打开转盘锁

221 阅读1分钟

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 之中。

每个 deadendstarget 中的字符串的数字会在 '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