leetCode 841. 钥匙和房间图遍历深度优先遍历+广度优先遍历 + 图解

呵呵哒(￣▽￣)"

已于 2023-11-27 12:05:37 修改

阅读量801

点赞数 12

分类专栏： leetCode 回溯算法文章标签：深度优先 leetcode 图遍历广度优先

于 2023-11-27 10:43:47 首次发布

本文链接：https://blog.csdn.net/weixin_41987016/article/details/134639679

版权

leetCode 同时被 2 个专栏收录

116 篇文章

订阅专栏

回溯算法

15 篇文章

订阅专栏

文章讲述了如何使用深度优先搜索和广度优先搜索算法解决LeetCode问题841，判断能否用钥匙进入所有房间。两种方法都以O(n+m)的时间复杂度，O(n)的空间复杂度实现。

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

841. 钥匙和房间 - 力扣（LeetCode）

有 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 号房间。

（1）图的深度优先遍历：

思路：使用深度优先搜索的方式遍历整张图，统计可以到达的节点个数，并利用数组 visit 记录当前节点是否访问过，避免重复访问

class Solution {
public:
    vector<bool> visit;
    int visCount=0;
    void dfs(vector<vector<int>>& rooms,int roomId){
        visit[roomId] = true;
        visCount++;
        for(auto& roomKey:rooms[roomId]) {
            if(!visit[roomKey]) {
                dfs(rooms,roomKey);
            }
        }
    }
    bool canVisitAllRooms(vector<vector<int>>& rooms) {
        int n = rooms.size();// 房间个数
        visit.resize(n);
        dfs(rooms,0);
        return visCount == n;
    }
};

分析复杂度：

时间复杂度：O(n+m)，其中 n 是房间的数量，m 是所有房间中的钥匙数量的总数
空间复杂度：O(n)，其中 n 是房间的数量，主要为栈的开销

（2）图的广度优先遍历：

思路：使用广度优先搜索的方式遍历整张图，统计可以到达的节点个数，并利用数组 visit 记录当前节点是否访问过，避免重复访问

class Solution {
public:
    bool canVisitAllRooms(vector<vector<int>>& rooms) {
        int n = rooms.size(),visCount=0;
        vector<bool> visit(n,0);
        visit[0]=true;
        queue<int> Q;
        Q.emplace(0);
        while(!Q.empty()) {
            int roomId = Q.front();
            Q.pop();
            visCount++;
            for(auto& roomKey : rooms[roomId]) {
                if(!visit[roomKey]) {
                    visit[roomKey] = true;
                    Q.emplace(roomKey);
                }
            }   
        }
        return visCount==n;
    }
};

分析复杂度：