标签：room dfs num 钥匙 rooms LeetCode841 id cur

一、题目

提示：

• 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

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。