【LeetCode841】钥匙和房间(dfs)

一、题目

【LeetCode841】钥匙和房间(dfs)
提示:

  • 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为栗子画了个图:
【LeetCode841】钥匙和房间(dfs)
像右边,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);            
        }
    }
};

【LeetCode841】钥匙和房间(dfs)
版本二:正确代码:

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);            
        }
    }
};

【LeetCode841】钥匙和房间(dfs)

上一篇:实现栈 lintcode495


下一篇:问题描述: abccba是一个特殊的字符串,正读反读都能读通的,这叫做回文字符串。 输入一个字符串判断是不是回文