Leetcode-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] 的值 互不相同

思路

设置一个辅助数组flags记录每个房间是否已经访问，然后DFS，访问过的设置成1，最后遍历flags数组，判断是否还存在没访问过的房间。

C++ Code

class Solution {
public:

    void dfs(int i,vector<vector<int>>& rooms,vector<int>& flags){
        vector<int> temp=rooms[i];;
        if(flags[i]==1) return;
        flags[i]=1;
        for(int a=0;a<temp.size();a++)
        {
            dfs(temp[a],rooms,flags);
        }

    }
    bool canVisitAllRooms(vector<vector<int>>& rooms) {
        vector<int> flags(rooms.size(),0);
        dfs(0,rooms,flags);
        for(int i=0;i<flags.size();i++)
        {
            if(flags[i]==0) return false;
        }
        return true;
    }
};