钥匙和房间 - Cocos3d-x Blog

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/keys-and-rooms
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

有 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 {
public:
    bool canVisitAllRooms(vector<vector<int>>& rooms) {
        std::vector<int> visited( rooms.size(), 0 );
        std::queue<int> que;
        que.push( 0 );
        visited[0] = 1;

        while( !que.empty() )
        {
            int room = que.front();
            que.pop();
            const vector<int>& keys = rooms[room];
            for ( unsigned int i = 0; i < keys.size(); i++ )
            {
                if ( !visited[keys[i]] )
                {
                    visited[keys[i]] = 1;
                    que.push( keys[i] );
                }
            }
        }

        for ( unsigned int i = 0; i < visited.size(); i++ )
        {
            if ( visited[i] == 0 )
                return false;
        }

        return true;
    }
};

本文由 BeijingJW 创作，采用知识共享署名4.0 国际许可协议进行许可
本站文章除注明转载/出处外，均为本站原创或翻译，转载前请务必署名
最后编辑时间为: Jun 2, 2023 at 05:21 pm