leetcode841 Keys and Rooms

"""
There are N rooms and you start in room 0.  Each room has a distinct number in 0, 1, 2, ..., N-1, and each room may have some keys to access the next room.
Formally, each room i has a list of keys rooms[i], and each key rooms[i][j] is an integer in [0, 1, ..., N-1] where N = rooms.length.  A key rooms[i][j] = v opens the room with number v.
Initially, all the rooms start locked (except for room 0).
You can walk back and forth between rooms freely.
Return true if and only if you can enter every room.
Example 1:
Input: [[1],[2],[3],[]]
Output: true
Explanation:
We start in room 0, and pick up key 1.
We then go to room 1, and pick up key 2.
We then go to room 2, and pick up key 3.
We then go to room 3.  Since we were able to go to every room, we return true.
Example 2:
Input: [[1,3],[3,0,1],[2],[0]]
Output: false
Explanation: We can't enter the room with number 2.
"""
"""
有两种方法，分别是BFS和DFS
解法一：BFS
用一个queue来存能到达的room
用一个set来存能拿到的房间钥匙
返回值为set里的钥匙数 和 房间数是否相等
"""
class Solution1:
    def canVisitAllRooms(self, rooms):
        queue = [0]
        s = set()
        s.add(0)  #!!!bug 没有初始化开0门的钥匙
        while queue:
            keys = queue.pop()
            for key in rooms[keys]:
                if key not in s:
                    s.add(key)
                    queue.append(key)
        return len(s) == len(rooms)

"""
解法二：DFS
建立一个set存钥匙，从rooms[0]开始递归
将找到的key存入set里
继续递归访问rooms[key]
"""
class Solution2:
    def canVisitAllRooms(self, rooms):
        s = set()
        s.add(0)
        def enterroom(keys):
            for key in keys:
                if key not in s:
                    s.add(key)
                    enterroom(rooms[key])
                # else:
                #     pass
            return
        enterroom(rooms[0])
        return len(s) == len(rooms)