【广度优先搜索-中等】841. 钥匙和房间

菜菜2022

于 2022-09-12 16:19:29 发布

阅读量125

点赞数

CC 4.0 BY-SA版权

分类专栏：刷题文章标签：宽度优先深度优先 leetcode

本文链接：https://blog.youkuaiyun.com/kz_java/article/details/126818556

刷题专栏收录该内容

643 篇文章

订阅专栏

本文探讨了三种方法解决房间探索问题，包括深度优先搜索（DFS）、普通模拟法和优先级队列的解决方案。通过实例展示了如何判断是否能访问所有房间，核心在于路径遍历策略和状态管理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【题目】
在这里插入图片描述
1 <= sum(rooms[i].length) <= 3000
0 <= rooms[i][j] < n
所有 rooms[i] 的值互不相同

【代码】
【方法1】深度优先搜索
在这里插入图片描述

class Solution:
    def canVisitAllRooms(self, rooms: List[List[int]]) -> bool:
        visited=[0 for i in range(len(rooms))]
        def dfs(x):
            visited[x]=1
            for item in rooms[x]:
                if visited[item]==0:
                    dfs(item)
        dfs(0)
        return sum(visited)==len(visited)

【方法2】普通模拟
在这里插入图片描述

class Solution:
    def canVisitAllRooms(self, rooms: List[List[int]]) -> bool:
        visited=[0 for i in range(len(rooms))]
        visited[0]=1
        keys=set(rooms[0])
        prelen=-1
        while prelen!=len(keys):
            prelen=len(keys)
            for key in keys:
                visited[key]=1
                keys=keys|set(rooms[key])
        return sum(visited)==len(visited)

【方法3】
在这里插入图片描述

class Solution:
    def canVisitAllRooms(self, rooms: List[List[int]]) -> bool:
        visited=[0 for i in range(len(rooms))]
        keys=[0]
        while keys:
            item=keys.pop(0)
            visited[item]=1
            for key in rooms[item]:
                if visited[key]==0:
                    keys.append(key)
        return sum(visited)==len(visited)