【LeetCode841】钥匙和房间(dfs)
一、题目
提示:
n == rooms.length2 <= n <= 10000 <= rooms[i].length <= 10001 <= sum(rooms[i].length) <= 30000 <= 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 visited; //已经访问过的房间个数 int cur_num = 0; int room_num;public: bool canVisitAllRooms(vector>& 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>& rooms, int room_id, int cur_num){ //递归边界判断 if(visited[room_id] || cur_num == room_num){ return; } //先访问当前的房间 //vectorid = 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 visited; //已经访问过的房间个数 int room_num;public: bool canVisitAllRooms(vector>& 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>& rooms, int room_id){ //递归边界判断 if(visited[room_id]){ return; } //先访问当前的房间 //vectorid = rooms[room_id]; visited[room_id] = 1; //遍历当前房间拥有的钥匙的房间 for(auto& next_i: rooms[room_id]){ dfs(rooms, next_i); } }};
版权声明:本文内容由网络用户投稿,版权归原作者所有,本站不拥有其著作权,亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容,请联系我们jiasou666@gmail.com 处理,核实后本网站将在24小时内删除侵权内容。
暂时没有评论,来抢沙发吧~