Leetcode 841. 钥匙和房间

news2024/11/30 2:36:11

1.题目基本信息

1.1.题目描述

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

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

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

1.2.题目地址

https://leetcode.cn/problems/keys-and-rooms/description/

2.解题方法

2.1.解题思路

图的深度优先搜索

2.2.解题步骤

第一步，构建有向图（rooms本质上就是邻接表形式的图，所以直接用即可）

第二步，递归任务：从node节点开始访问图中的节点，并使用visited集合记录访问的节点

第三步，根据访问的节点个数判断是否可以访问所有的房间

3.解题代码

Python代码

class Solution:
    def canVisitAllRooms(self, rooms: List[List[int]]) -> bool:
        length=len(rooms)
        # 第一步，构建有向图（rooms本质上就是邻接表形式的图，所以直接用即可）
        graph=rooms # 本质上就是一个邻接表
        visited=set()
        # 第二步，递归任务：从node节点开始访问图中的节点，并使用visited集合记录访问的节点
        def dfs(node):
            if node in visited:
                return
            visited.add(node)
            for subNode in graph[node]:
                dfs(subNode)
        dfs(0)
        # print(visited)
        # 第三步，根据访问的节点个数判断是否可以访问所有的房间
        return len(visited)==length

C++代码

class Solution {
public:
    set<int> visited;
    void dfs(int node,vector<vector<int>>& graph){
        if(visited.find(node)!=visited.end()){
            return;
        }
        visited.insert(node);
        for(int subNode:graph[node]){
            dfs(subNode,graph);
        }
    }

    bool canVisitAllRooms(vector<vector<int>>& rooms) {
        int length=rooms.size();
        dfs(0,rooms);
        return visited.size()==length;
    }
};