BFS广度优先搜索之钥匙和房间

短命女 2023-09-30 22:21 2阅读 0赞

[841. 钥匙和房间][841.]

有 n 个房间，房间按从 0 到 n - 1 编号。最初，除 0 号房间外的其余所有房间都被锁住。你的目标是进入所有的房间。然而，你不能在没有获得钥匙的时候进入锁住的房间。
    
    当你进入一个房间，你可能会在里面找到一套不同的钥匙，每把钥匙上都有对应的房间号，即表示钥匙可以打开的房间。你可以拿上所有钥匙去解锁其他房间。
    
    给你一个数组 rooms 其中 rooms[i] 是你进入 i 号房间可以获得的钥匙集合。如果能进入 所有 房间返回 true，否则返回 false。
    
    示例 1：
    
    输入：rooms = [[1],[2],[3],[]]
    输出：true
    解释：
    我们从 0 号房间开始，拿到钥匙 1。
    之后我们去 1 号房间，拿到钥匙 2。
    然后我们去 2 号房间，拿到钥匙 3。
    最后我们去了 3 号房间。
    由于我们能够进入每个房间，我们返回 true。
    示例 2：
    
    输入：rooms = [[1,3],[3,0,1],[2],[0]]
    输出：false
    解释：我们不能进入 2 号房间。
    
    提示：
    
    n == rooms.length
    2 <= n <= 1000
    0 <= rooms[i].length <= 1000
    1 <= sum(rooms[i].length) <= 3000
    0 <= rooms[i][j] < n
    所有 rooms[i] 的值 互不相同

题解：

class Solution:
        def canVisitAllRooms(self, rooms: List[List[int]]) -> bool:
            visited = {
        0}	# 记录已经穷举过的钥匙，防止走回头路
            queue = [0]
    
            while queue:
                index_keys = queue.pop()
                for i in rooms[index_keys]:
                    if i not in visited:
                        visited.add(i)
                        queue.insert(0, i)
    
            return len(visited) == len(rooms)

这个题还可以用 `DFS`，只需将 `queue.insert` 换成 `queue.append` 即可

> 参考：[7行DFS 8行BFS 两种方法 三种实现 超详细趣味０基础解 Python][7_DFS 8_BFS _ _ _ Python]

[841.]: https://leetcode.cn/problems/keys-and-rooms/
[7_DFS 8_BFS _ _ _ Python]: https://leetcode.cn/problems/keys-and-rooms/solution/7xing-dfs-8xing-bfs-liang-chong-fang-fa-san-chong-/