leetcode 841. 钥匙和房间

1. 题目链接 https://leetcode-cn.com/problems/keys-and-rooms/comments/

2. 题目描述

   1. 有 N 个房间，开始时你位于 0 号房间。每个房间有不同的号码：0，1，2，...，N-1，并且房间里可能有一些钥匙能使你进入下一个房间。

      在形式上，对于每个房间 i 都有一个钥匙列表 rooms[i]，每个钥匙 rooms[i][j] 由 [0,1，...，N-1] 中的一个整数表示，其中 N = rooms.length。 钥匙 rooms[i][j] = v 可以打开编号为 v 的房间。

      最初，除 0 号房间外的其余所有房间都被锁住。

      你可以自由地在房间之间来回走动。

      如果能进入每个房间返回 true，否则返回 false。

   2. 示例 1：

      输入: [[1],[2],[3],[]]
      输出: true
      解释:  
      我们从 0 号房间开始，拿到钥匙 1。
      之后我们去 1 号房间，拿到钥匙 2。
      然后我们去 2 号房间，拿到钥匙 3。
      最后我们去了 3 号房间。
      由于我们能够进入每个房间，我们返回 true。
      

      示例 2：

      输入：[[1,3],[3,0,1],[2],[0]]
      输出：false
      解释：我们不能进入 2 号房间。
      

3. 解题思路

   1. 判断连通图，dfs或是bfs都可以，考虑到回路问题，要有一个visit数组来判断节点是否访问过

4. 代码

   1. python(bfs)

      class Solution:
          def canVisitAllRooms(self, rooms):
              from collections import deque
      
              vis = [0] * len(rooms)
              vis[0] = 1
              q = deque(rooms[0])
              while q:
                  key = q.popleft()
                  q += (1 - vis[key]) * rooms[key]
                  vis[key] = 1
      
              return sum(vis) == len(rooms)
      

   2. python(dfs)

      class Solution:
          def canVisitAllRooms(self, rooms):
              vis = [0] * len(rooms)
              vis[0] = 1
              
              def _dfs(i):
                  nonlocal vis
                  for j in rooms[i]:
                      if not vis[j]:
                          vis[j] = 1
                          _dfs(j)
              
              _dfs(0)
              return sum(vis) == len(rooms)