友情支持

如果您觉得这个笔记对您有所帮助,看在D瓜哥码这么多字的辛苦上,请友情支持一下,D瓜哥感激不尽,😜

支付宝

微信

有些打赏的朋友希望可以加个好友,欢迎关注D 瓜哥的微信公众号,这样就可以通过公众号的回复直接给我发信息。

wx jikerizhi

公众号的微信号是: jikerizhi因为众所周知的原因,有时图片加载不出来。 如果图片加载不出来可以直接通过搜索微信号来查找我的公众号。

841. 钥匙和房间

n 个房间,房间按从 0n - 1 编号。最初,除 0 号房间外的其余所有房间都被锁住。你的目标是进入所有的房间。然而,你不能在没有获得钥匙的时候进入锁住的房间。

当你进入一个房间,你可能会在里面找到一套 不同的钥匙,每把钥匙上都有对应的房间号,即表示钥匙可以打开的房间。你可以拿上所有钥匙去解锁其他房间。

给你一个数组 rooms 其中 rooms[i] 是你进入 i 号房间可以获得的钥匙集合。如果能进入 所有 房间返回 true,否则返回 false

示例 1:

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

示例 2:

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

提示:

  • n == rooms.length

  • 2 <= n <= 1000

  • 0 <= rooms[i].length <= 1000

  • 1 <= sum(rooms[i].length) <= 3000

  • 0 <= rooms[i][j] < n

  • 所有 rooms[i] 的值 互不相同

思路分析

根据钥匙,把房间走一遍即可。可以使用备忘录来防止重复走同一个房间来加快速度。

既可以深度优先遍历,也可以广度优先遍历。如果使用深度优先遍历,需要使用递归。

  • 一刷

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
/**
 * @author D瓜哥 · https://www.diguage.com
 * @since 2025-05-24 18:48:35
 */
public boolean canVisitAllRooms(List<List<Integer>> rooms) {
  Queue<Integer> queue = new LinkedList<>();
  queue.addAll(rooms.get(0));
  rooms.set(0, null);
  while (!queue.isEmpty()) {
    Integer key = queue.poll();
    List<Integer> keys = rooms.get(key);
    if (keys != null) {
      queue.addAll(keys);
      rooms.set(key, null);
    }
  }
  return rooms.stream().allMatch(Objects::isNull);
}