LeetCode笔记（841.钥匙和房间）

合集 - LeetCode笔记(1)

1.LeetCode笔记（841.钥匙和房间）2024-02-20

收起

LeetCode 841.钥匙和房间

题目描述

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

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

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

示例

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

输入：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. 定义两个list，分别是todo记录未进入的房间，done记录已进入的房间
2. 如果todo_list不为空，则取一个值为当前房间（初始化为[0]）
3. 当前房间的钥匙号，如果在done_list、todo_list里面，则无需处理，否则放进todo_list
4. 一直循环处理todo_list直至为空
5. 此时done_list的长度应与rooms一致，否则返回False

class Solution:
    def canVisitAllRooms(self, rooms: List[List[int]]) -> bool:
        todo_list, done_list = [0], []
        next_rooms = []

        while todo_list:
            cur_room = todo_list[0]
            for i in rooms[cur_room]:
                next_rooms.append(i)
            for j in next_rooms:
                if j in todo_list or j in done_list:
                    pass
                else:
                    todo_list.append(j)

            # 当前房间已处理，放进done_list，并从todo_list移除当前房间
            # 重置当前房间的钥匙列表
            done_list.append(cur_room)
            todo_list.pop(0)
            next_rooms.clear()

        return len(done_list) == len(rooms)