aboutsummaryrefslogtreecommitdiff
path: root/other/keys_and_rooms.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'other/keys_and_rooms.cpp')
-rw-r--r--other/keys_and_rooms.cpp32
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";
+}