一、题目

提示：

• n == rooms.length
• 2 <= n <= 1000
• 0 <= rooms[i].length <= 1000
• 1 <= sum(rooms[i].length) <= 3000
• 0 <= rooms[i][j] < n
• 所有 rooms[i] 的值 互不相同

二、思路

dfs基础题。以测试用例2为栗子画了个图：

像右边，3号为父结点和孩子结点之间是不连接的，因为0号房间已经遍历过了。
一开始用版本一的代码有些测试用例没过，找了好久是因为dfs的递归边界和访问时的操作出了问题，不应该简单用cur_num（当前已经访问过的房间种数）进行判断是否完成任务，想法上没问题，但是我设置的cur_num是全局变量，遍历多条路径时都会依次累加这个cur_num值就不对啦，所以可以直接判断哈希表visited的size是否为房间总数即可—>作为最后成功与否的判断；并且也不用cur_num了。

当然从时间复杂度的结果看，还可进一步剪枝。

三、代码

版本一：错误代码（只过了48个测试用例）：

class Solution {
private:
    unordered_map<int, int> visited;
    //已经访问过的房间个数
    int cur_num = 0;
    int room_num;
public:
    bool canVisitAllRooms(vector<vector<int>>& rooms) {
        if(rooms.size() == 0){
            return false;
        }
        int room_num = rooms.size();
        dfs(rooms, 0, cur_num);
        if(room_num == cur_num){
            return true;
        }else{
            return false;
        }
    }
    //cur_num为当前已经访问过的房间个数（种类）
    void dfs(vector<vector<int>>& rooms, int room_id, int cur_num){
        //递归边界判断
        if(visited[room_id] || cur_num == room_num){   
            return;
        }
        //先访问当前的房间
        //vector<int>id = rooms[room_id];
        visited[room_id] = 1;
        cur_num++; 

        //遍历当前房间拥有的钥匙的房间
        for(auto& next_i: rooms[room_id]){
            dfs(rooms, next_i, cur_num);            
        }
    }
};

版本二：正确代码：

class Solution {
private:
    unordered_map<int, int> visited;
    //已经访问过的房间个数
    int room_num;
public:
    bool canVisitAllRooms(vector<vector<int>>& rooms) {
        if(rooms.size() == 0){
            return false;
        }
        int room_num = rooms.size();
        dfs(rooms, 0);
        if(room_num == visited.size()){
            return true;
        }else{
            return false;
        }
    }
    //cur_num为当前已经访问过的房间个数（种类）
    void dfs(vector<vector<int>>& rooms, int room_id){
        //递归边界判断
        if(visited[room_id]){   
            return;
        }
        //先访问当前的房间
        //vector<int>id = rooms[room_id];
        visited[room_id] = 1;

        //遍历当前房间拥有的钥匙的房间
        for(auto& next_i: rooms[room_id]){
            dfs(rooms, next_i);            
        }
    }
};

标签：room,dfs,num,钥匙,rooms,LeetCode841,id,cur
来源： https://blog.csdn.net/qq_35812205/article/details/122529205