1.23 力扣图论

en7oy2024-01-24 14:36

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

思路：本题其实给我们是一个有向图，不必做转换

用bfs方法，从第一个房间开始，拿钥匙、进入另一个房间、拿钥匙。。。这样循环，并最后判断是否所有房间都进入过了

cpp 复制代码

class Solution {
public:
    bool canVisitAllRooms(vector<vector<int>>& rooms) {
        if(rooms.empty()) return true;
        queue<int> q;
        //visitRooms标记对应房间是否进入过了
        vector<bool> visitRooms(rooms.size(),false);
        q.push(0);
        while(!q.empty())
        {
            vector<int> keyRooms=rooms[q.front()];
            visitRooms[q.front()]=true;
            q.pop();
            for(int keyRoom:keyRooms)
            {
                if(!visitRooms[keyRoom]) q.push(keyRoom);
            }
        }
        for(bool visited:visitRooms)
        {
            if(!visited) return false;
        }
        return true;
    }
};