leetcode-841-鑰匙和房間

題目描述

有 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。

提示：

1 <= rooms.length <= 1000
0 <= rooms[i].length <= 1000
所有房間中的鑰匙數量總計不超過 3000。

示例

• 示例 1:

  輸入: [[1],[2],[3],[]]
  輸出: true
  解釋:  
  我們從 0 號房間開始，拿到鑰匙 1。
  之後我們去 1 號房間，拿到鑰匙 2。
  然後我們去 2 號房間，拿到鑰匙 3。
  最後我們去了 3 號房間。
  由於我們能夠進入每個房間，我們返回 true。
  

• 示例 2:

  輸入：[[1,3],[3,0,1],[2],[0]]
  輸出：false
  解釋：我們不能進入 2 號房間。
  

  來源：力扣（LeetCode）
  連結：https://leetcode-cn.com/problems/robot-return-to-origin

---

雖然歸類為中等題型，但是意外地簡單。類似鏈式結構，房間代表連結串列的節點，鑰匙代表指向的下一節點。

首先想到的就是遞迴， 構造遞迴函式lock, 開鎖即進入房間。沒去過的房間進行遞迴操作，否則不重複進入。

最後， 根據房間的數量和已使用鑰匙集合的數量判斷，若房間數大於鑰匙數，則有沒去過的房間， 返回False.

• 題解1:

  執行用時：88ms,在所有Python3提交中擊敗了53.68%的使用者

  記憶體消耗：14.5MB,在所有Python3提交中擊敗了13.44%的使用者

  class Solution:
  def canVisitAllRooms(self, rooms) -> bool:
      used_key = set()
      used_key.add(0)
      s_len = len(rooms)
  
      def lock(keys):
          if keys:
              for key in keys:
                  if key in used_key:
                      continue
                  else:
                      used_key.add(key)
                      lock(rooms[key])
  
      for index, room in enumerate(rooms):
          if room and index in used_key:
              lock(room)
  
      return True if s_len <= len(used_key) else False