Keys and Rooms

841. Keys and Rooms

@@ -0,0 +1,26 @@
+from queue import Queue
+
+
+class Solution:
+    def canVisitAllRooms(self, rooms):
+        """
+        :type rooms: List[List[int]]
+        :rtype: bool
+        """
+        s=set(rooms[0])
+        s.add(0)
+        q=Queue()
+        for k in rooms[0]:
+            q.put(k)
+        while not q.empty():
+            mask=q.get()
+            for k in rooms[mask]:
+                if k not in s:
+                    s.add(k)
+                    q.put(k)
+
+        return len(s)==len(rooms)
+
+rooms=[[1,3],[3,0,1],[2],[0]]
+sol=Solution()
+print(sol.canVisitAllRooms(rooms))