友情支持
如果您觉得这个笔记对您有所帮助,看在D瓜哥码这么多字的辛苦上,请友情支持一下,D瓜哥感激不尽,😜
有些打赏的朋友希望可以加个好友,欢迎关注D 瓜哥的微信公众号,这样就可以通过公众号的回复直接给我发信息。
公众号的微信号是: jikerizhi 。因为众所周知的原因,有时图片加载不出来。 如果图片加载不出来可以直接通过搜索微信号来查找我的公众号。 |
841. 钥匙和房间
有 n
个房间,房间按从 0
到 n - 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);
}