打开转盘锁 -- BFS

打开转盘锁

这里提供两种实现,单向BFS和双向BFS。

python 复制代码
class OpenLock:
    """
    752. 打开转盘锁
    https://leetcode.cn/problems/open-the-lock/
    """
    def solution(self, deadends: List[str], target: str) -> int:
        """
        单向BFS
        :param deadends: 
        :param target: 
        :return: 
        """
        visited = set()
        # 将deadends初始化到visited数组中
        for deadend in deadends:
            visited.add(deadend)

        queue = []
        step = 0
        queue.append('0000')
        visited.add('0000')

        while queue:
            sz = len(queue)
            for i in range(sz):
                cur = queue.pop(0)

                if cur == target:
                    return step

                if cur in visited:
                    continue

                visited.add(cur)

                for j in range(4):
                    up = self.plusOne(cur, j)
                    if up not in visited:
                        queue.append(up)

                    down = self.minusOne(cur, j)
                    if down not in visited:
                        queue.append(down)

            step += 1
        return -1

    def plusOne(self, cur, j):
        if cur[j] == '9':
            return cur[:j] + '0' + cur[j+1:]
        else:
            return cur[:j] + str(int(cur[j])+1) + cur[j+1:]

    def minusOne(self, cur, j):
        if cur[j] == '0':
            return cur[:j] + '9' + cur[j + 1:]
        else:
            return cur[:j] + str(int(cur[j]) - 1) + cur[j + 1:]

    def solution2(self, deadends: List[str], target: str) -> int:
        """
        双向BFS优化
        :param deadends: 
        :param target: 
        :return: 
        """
        deads = set()
        visited = set()
        # 将deadends初始化到visited数组中
        for deadend in deadends:
            deads.add(deadend)

        q1 = set()
        q2 = set()
        q1.add('0000')
        q2.add(target)
        step = 0

        while q1 and q2:
            # 额外优化
            if len(q1) > len(q2):
                tmp = q1
                q1 = q2
                q2 = tmp

            temp = set()
            for cur in q1:
                if cur in deads:
                    continue
                if cur in q2:
                    return step

                visited.add(cur)

                for j in range(4):
                    up = self.plusOne(cur, j)
                    if up not in visited:
                        temp.add(up)

                    down = self.minusOne(cur, j)
                    if down not in visited:
                        temp.add(down)

            step += 1
            # 这里temp相当于q1,交换q1和q2
            q1 = q2
            q2 = temp

        return -1
相关推荐
一只鱼^_2 天前
基础算法合集-图论
数据结构·算法·深度优先·图论·广度优先·宽度优先·图搜索算法
reverie.Ly17 天前
宽度优先遍历(bfs)(2)——fllodfill算法
算法·宽度优先
小猫咪怎么会有坏心思呢18 天前
华为OD机试真题-矩阵扩散-BFS(JAVA)
华为od·矩阵·宽度优先
青山是哪个青山21 天前
位运,模拟,分治,BFS,栈和哈希表
算法·散列表·宽度优先
a东方青22 天前
[蓝桥杯 2023 国 B] AB 路线 (BFS)
c++·算法·职场和发展·蓝桥杯·宽度优先
飞川撸码1 个月前
【LeetCode 热题100】BFS/DFS 实战:岛屿数量 & 腐烂的橘子(力扣200 / 994 )(Go语言版)
leetcode·深度优先·宽度优先
wen__xvn1 个月前
BFS入门刷题
算法·宽度优先
Swift社区1 个月前
面试高频图论题『墙与门』:Swift BFS 解法全流程拆解
面试·swift·宽度优先
图初1 个月前
路径规划算法BFS/Astar/HybridAstar简单实现
算法·动态规划·astar·宽度优先·hybridastar
算法给的安全感2 个月前
bfs-最小步数问题
java·算法·宽度优先