diff options
Diffstat (limited to 'other/keys_and_rooms.cpp')
-rw-r--r-- | other/keys_and_rooms.cpp | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/other/keys_and_rooms.cpp b/other/keys_and_rooms.cpp new file mode 100644 index 0000000..59308ef --- /dev/null +++ b/other/keys_and_rooms.cpp @@ -0,0 +1,32 @@ +#include "leetcode.h" + +class Solution { +public: + bool canVisitAllRooms(vvd & rooms) { + stack<int> dfs; + dfs.push(0); + unordered_set<int> seen = {0}; + while (!dfs.empty()) { + int i = dfs.top(); + dfs.pop(); + for (int j : rooms[i]) { + if (seen.count(j) == 0) { + dfs.push(j); + seen.insert(j); + if (rooms.size() == seen.size()) + return true; + } + } + } + return rooms.size() == seen.size(); + } +}; + +int main() { + Solution obj; + vvd rooms = {{1}, {2}, {3}, {}}; + if (obj.canVisitAllRooms(rooms)) + cout << "true"; + else + cout << "false"; +} |