【Warrior刷题笔记】LC841. 钥匙和房间 【dfs】详细注释简单易懂

题目

LC841.钥匙和房间

解题思路

本题可以使用深度优先遍历解决。
我们使用visited数组值visited[i]标记每个房间i是否遍历过，防止重复遍历。然后从房间i=0开始，遍历拿到的钥匙rooms[i][j]所能去到的所有房间，并将遍历过的房间的已遍历标志位visited[i]置1，同时在第一次遍历某房间时将已遍历房间数count+1，在遍历完整个数组后判断count是否等于数组长m，若相等则返回真，否则返回假。

代码

class Solution {
public:
    bool canVisitAllRooms(vector<vector<int>>& rooms) {
        int m = rooms.size();//计算数组长
        vector<int> visited(m,0);//记录房间是否被遍历过
        int count = 0;//记录已遍历房间数
        function<void(int)> dfs = [&](int pos){//dfs
            if(visited[pos]==0){//若未遍历过
                ++visited[pos];//已遍历标志位置1
                ++count;//已遍历房间数加一
                if(count==m) return;//如果已遍历完所有房间直接返回
                for(const auto& key : rooms[pos]) dfs(key);//遍历该房间取到的钥匙能开的所有房间
            }
            else return;//若当前房间已遍历过，直接返回
        };
        dfs(0);//从房间0开始dfs
        return count == m;//若count==m返回真
    }
};

复杂度分析

时间复杂度： O(m)。m为房间数，我们最多遍历每个房间各一次。
空间复杂度： O(m)。已访问标志数组的空间消耗。