lc841. Keys and Rooms

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))