有 n 个房间,房间按从 0 到 n - 1 编号。最初,除 0 号房间外的其余所有房间都被锁住。你的目标是进入所有的房间。然而,你不能在没有获得钥匙的时候进入锁住的房间。
当你进入一个房间,你可能会在里面找到一套不同的钥匙,每把钥匙上都有对应的房间号,即表示钥匙可以打开的房间。你可以拿上所有钥匙去解锁其他房间。
给你一个数组 rooms 其中 rooms[i] 是你进入 i 号房间可以获得的钥匙集合。如果能进入 所有 房间返回 true,否则返回 false。
思路: 用yaoshi栈记录每个房间有的钥匙,用visit表示访问过的房间,当访问的房价没有钥匙,并且手上没有钥匙的时候说明没办法继续访问房间了,最后如果visit访问了n个房间,就说明访问完了,否则就是没访问玩
class Solution {
public:
bool canVisitAllRooms(vector<vector<int>>& rooms) {
int n = rooms.size();
stack<int> yaoshi;
unordered_set<int> visit;
int i = 0;
yaoshi.push(0);
visit.insert(0);
while(!yaoshi.empty()){
i = yaoshi.top();
yaoshi.pop();
for(auto& j : rooms[i]){
if (visit.find(j) == visit.end()) {
yaoshi.push(j);
visit.insert(j);
}
}
}
return visit.size() == n;
}
};