第26天,回溯part04,昨天休息复习总结回溯内容,💪(ง •_•)ง💪
目录
491.递增子序列
46.全排列
47.全排列 II
51.N皇后
37.解数独
回溯总结
491.递增子序列
文档讲解:代码随想录递增子序列
视频讲解:手撕递增子序列
题目:
学习:本题同属于子集问题,但不同的在于给了多个限制:1.子序列必须包含两个元素以上;2.子序列必须是递增的;3.不能对数组进行排序,即不能改变各元素之间的位置关系。
同时本题还需要进行去重,由于不能对数组进行排序,因此不能采用之前的去重办法,可以通过设置一个哈希表的方式进行去重。
代码:
//时间复杂度O(n*2^n)
//空间复杂度O(n)
class Solution {
public:
vector<vector<int>> result; //返回数组
vector<int> path; //记录路径
//确定返回值和参数列表,在返回数组中直接进行操作所以不需要返回值,设置startindex确定每一层遍历范围
void backtracking (vector<int>& nums, int startindex) {
//收取路径大于1的节点
if(path.size() > 1) {
result.push_back(path);
}
//确定返回条件(在子集问题中可以不写)因为当startindex==nums.size()时,不会进入下面的循环
if(startindex == nums.size()) {
return;
}
//确定单层递归逻辑
unordered_set<int> used; //使用哈希表记录已经使用的数字
for(int i = startindex; i < nums.size(); i++) {
//查找当前遍历的数字是否符合要求: 1.当前数字是否大于path中最后一个数字;2.当前数字是否没有遍历过
if(!path.empty() && nums[i] < path.back() || (used.find(nums[i]) != used.end())) {
continue; //进行下一轮循环
}
used.insert(nums[i]);
//复合我们需要的数字条件
path.push_back(nums[i]);
backtracking(nums, i + 1);
path.pop_back(); //回溯
}
}
vector<vector<int>> findSubsequences(vector<int>& nums) {
backtracking(nums, 0);
return result;
}
};
注意:本题规定了数值的范围在[-100,100]之间,数量一定且有序,因此本题也可以使用数组作为哈希表,能够提高算法效率。程序运行的时候对unordered_set 频繁的insert,unordered_set需要做哈希映射(也就是把key通过hash function映射为唯一的哈希值)相对费时间,而且每次重新定义set,insert的时候其底层的符号表也要做相应的扩充,也是费事的。
代码:
class Solution {
public:
vector<vector<int>> result; //返回数组
vector<int> path; //记录路径
//确定返回值和参数列表,在返回数组中直接进行操作所以不需要返回值,设置startindex确定每一层遍历范围
void backtracking (vector<int>& nums, int startindex) {
//收取路径大于1的节点
if(path.size() > 1) {
result.push_back(path);
}
//确定返回条件(在子集问题中可以不写)因为当startindex==nums.size()时,不会进入下面的循环
if(startindex == nums.size()) {
return;
}
//确定单层递归逻辑
int used[201] = {0}; //使用数组来记录数字是否使用过,因为题干规定数字在[-100, 100]之间。
for(int i = startindex; i < nums.size(); i++) {
//查找当前遍历的数字是否符合要求: 1.当前数字是否大于path中最后一个数字;2.当前数字是否没有遍历过
if(!path.empty() && nums[i] < path.back() || (used[nums[i] + 100] == 1)) {
continue; //进行下一轮循环
}
used[nums[i] + 100] = 1;
//复合我们需要的数字条件
path.push_back(nums[i]);
backtracking(nums, i + 1);
path.pop_back(); //回溯
}
}
vector<vector<int>> findSubsequences(vector<int>& nums) {
backtracking(nums, 0);
return result;
}
};
46.全排列
文档讲解:代码随想录全排列
视频讲解:手撕全排列
题目:
学习:回溯算法解决第四类排列问题,排列问题的特点在于:1.在叶子结点收获结果;2.每层循环需从i = 0开始。因为对于排列问题来说[1,2]与[2,1]是不同的排列方式。
基于以上两点,排列不能使用startindex确定循环范围,而是需要一个used数组来标识每个元素是否被使用。
代码:
//时间复杂度O(n!)
//空间复杂度O(n)
class Solution {
public:
//排列问题
vector<vector<int>> result; //返回数组
vector<int> path; //记录路径
//确定返回值和参数列表,在返回数组中直接操作,因此不需要返回值,与组合问题不同,排列存在顺序因此需要用一个数组来记录循环范围
void backtracking(vector<int>& nums, vector<bool>& used) {
//确定终止条件,在叶子节点处收获结果
if(path.size() == nums.size()) {
result.push_back(path);
return;
}
//确定单层递归逻辑,每次都得从0开始,由used数组来确定遍历的范围
for(int i = 0; i < nums.size(); i++) {
//如果当前数遍历过,则跳过继续往后遍历
if(used[i] == true) {
continue;
}
//如果没有遍历过,进入单层递归逻辑
used[i] = true;
path.push_back(nums[i]);
backtracking(nums, used);
//回溯
path.pop_back();
used[i] = false;
}
}
vector<vector<int>> permute(vector<int>& nums) {
//设置一个bool类型的数组,记录每次遍历过程中该数有没有被使用,事实上这种方法,在前面的去重中也能够使用
vector<bool> used(nums.size(), false);
backtracking(nums, used);
return result;
}
};
47.全排列 II
文档讲解:代码随想录全排列II
视频讲解:手撕全排列II
题目:
学习:本题与上一题不同在于,本题存在重复数字需要进行去重处理。由于我们设置了used数组因此可以直接使用used数组进行去重处理。原理在于:当后面出现与前面相同的数字时,且前面的数字没有被使用(used == false),此时跳过当前数字。因为for循环顺序的原因,前面的数字一定先完成所有可能性的遍历,因此遍历到后面的数字,前面的数字没有被使用,说明已经遍历完成了,这种去重也属于树层去重。(注意:使用这种方式,需要对数组先进行排序)
代码:
//时间复杂度O(n!*n)
//空间复杂度O(n)
class Solution {
public:
//本题与前一题的不同之处在于需要进行去重
vector<vector<int>> result; //返回数组
vector<int> path; //记录路径
//确定返回值和参数列表,本题与上一题不同在于去重,但使用used数组就能够进行去重处理,因此参数列表相同
void backtracking(vector<int>& nums, vector<bool>& used) {
//确定终止条件
if(path.size() == nums.size()) {
result.push_back(path);
return;
}
//确定单层递归逻辑
for(int i = 0; i < nums.size(); i++) {
//首先进行数层去重,如果后面的数和前面的数相同,且前面的数已经完成了遍历回溯为false,则跳过
if(i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) {
continue;
}
//接着判断当前遍历的数是否被使用
if(used[i] == true) {
continue;
}
//进入单层递归逻辑
used[i] = true;
path.push_back(nums[i]);
backtracking(nums, used);
//回溯
path.pop_back();
used[i] = false;
}
}
vector<vector<int>> permuteUnique(vector<int>& nums) {
//建立一个used数组,记录数是否使用
vector<bool> used(nums.size(), false);
//对nums排序便于去重
sort(nums.begin(), nums.end());
backtracking(nums, used);
return result;
}
};
51.N皇后
文档讲解:代码随想录N皇后
视频讲解:手撕N皇后
题目:
学习: 回溯算法解决第五类棋盘问题。棋盘问题的特点在于需要遍历的不再是一维数组,而是需要遍历二维数组。但本质仍是一层递归一层循环,对于本题来说每层循环需要在一行插入一个皇后,同时插入的皇后有三个约束条件:1.不能同行;2.不能同列;3.不能同斜线。基于此也可以将棋盘问题的搜索过程抽象为一棵树:
注意:棋盘问题一般也是在叶子结点收获结果,由于存在插入是否可行的判断,因此只要搜索到了树的叶子结点,就说明找到了皇后们的合理位置了。
代码:本题关键在于理解棋盘的宽度就是for循环的长度,递归的深度就是棋盘的高度,这样就可以套进回溯法的模板里了。
//时间复杂度O(n!)
//空间复杂度O(n)
class Solution {
public:
//棋盘问题
vector<vector<string>> result; //返回数组
vector<string> chessboard; //棋盘也是每一个结果
//确定返回值和参数列表,循环遍历的是列,因此加入n和当前遍历的行
void backtracking(int n, int row) {
//确定终止条件,在叶子结点收获结果,能够走到叶子结点说明一定正确
if(row == n) {
result.push_back(chessboard);
return;
}
//确定单层递归逻辑,对列进行循环遍历
for(int i = 0; i < n; i++) {
//判断当前列能不能插入皇后
if(isvalid(n, row, i)) { //如果可以的话,进行下一层递归
chessboard[row][i] = 'Q';
backtracking(n, row + 1);
//回溯
chessboard[row][i] = '.';
}
}
}
//参数分别表示n,当前行数,当前列数
bool isvalid(int n, int row, int col) {
//判断是否冲突:注意本题不需要判断行是否冲突,因为在for循环遍历过程中我们只允许一行有一个皇后
//判断列是否冲突,后面的行还没有插入皇后,所以只需要遍历到row就可以
for(int i = 0; i < row; i++) {
if(chessboard[i][col] == 'Q') return false;
}
//判断45°角位置有没有皇后
for(int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
if(chessboard[i][j] == 'Q') return false;
}
//判断135°角位置有没有皇后
for(int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {
if(chessboard[i][j] == 'Q') return false;
}
return true;
}
vector<vector<string>> solveNQueens(int n) {
//对chessboard进行初始化处理
for(int i = 0; i < n; i++) {
chessboard.push_back(string(n,'.'));
}
backtracking(n, 0);
return result;
}
};
37.解数独
文档讲解:代码随想录解数独
视频讲解:手撕解数独
题目:
学习:本题也属于棋盘问题,本题也需要在二维数组中插入元素,并且不断判断插入的是否合理。 但与N皇后不同的在于,本题一行不仅仅插入一个元素,而是需要插入多个元素,需要对行和列都进行遍历。因此本题一层递归不止一个循环,一层递归包含两个循环,因此本题要做的是二维递归。
本题中棋盘的每一个位置都要放一个数字(而N皇后是一行只放一个皇后),并检查数字是否合法,解数独的树形结构要比N皇后更宽更深。用部分树来表示为:
注意:本题不需要遍历所有的结果,只需要找到一个结果就可以返回,因此本题的返回值不再是void,而是bool,这样能够在找到一个答案之后立马进行返回。
代码:
class Solution {
public:
//棋盘问题——二维回溯
char arr[9] = {'1','2','3','4','5','6','7','8','9'}; //设置1-9的字符数组方便遍历
//确定返回值和参数列表,本题返回值需要为bool类型,因为,本题不需要遍历所有的结果,只要找到一个结果就返回即可,因此选择bool类型
bool backtracking(vector<vector<char>>& board) {
//确定终止条件,本题需要终止条件,因为本题中间会遍历错误会进行返回,如果遍历到最后也无需判断当前情况。
//确定单层递归逻辑——两层循环
//遍历行
for(int i = 0; i < board.size(); i++) {
//遍历列
for(int j = 0; j < board.size(); j++) {
if(board[i][j] != '.') continue; //找到第一个空格点
//进行插入数字尝试
for(char k : arr) {
//判断当前位置是否可以插入
if(isvalid(board, i, j, k)) { //可以插入的话进入下一层循环
board[i][j] = k;
//当后面的结果回传回来
if(backtracking(board)) return true;
board[i][j] = '.'; //回溯
}
}
//当前位置1-9都不行,说明无法得到答案
return false;
}
}
//完成所有的遍历
return true;
}
//判断函数,参数列表分别为,原数组,行,列,插入的元素
bool isvalid(vector<vector<char>>& board, int row, int col, char val) {
//判断行是否重复
for(int j = 0; j < board.size(); j++) {
if(board[row][j] == val) return false;
}
//判断列是否重复
for(int i = 0; i < board.size(); i++) {
if(board[i][col] == val) return false;
}
//判断九宫格内是否有重复
//确定九宫格位置
int startrow = row / 3 * 3, startcol = col / 3 * 3;
for(int i = startrow; i < startrow + 3; i++) {
for(int j = startcol; j < startcol + 3; j++) {
if(board[i][j] == val) return false;
}
}
return true;
}
void solveSudoku(vector<vector<char>>& board) {
backtracking(board);
}
};
回溯总结
文档讲解:代码随想录回溯总结
回溯算法本质是一个暴力搜索的算法,并不是什么高效的算法。
回溯是递归的副产品,只要有递归就会有回溯,所以回溯法也经常和二叉树遍历,深度优先搜索混在一起,因为这两种方式都是用了递归。
回溯算法一般能够解决5类问题:
- 组合问题:N个数里面按一定规则找出k个数的集合
- 排列问题:N个数按一定规则全排列,有几种排列方式
- 切割问题:一个字符串按一定规则有几种切割方式
- 子集问题:一个N个数的集合里有多少符合条件的子集
- 棋盘问题:N皇后,解数独等等
回溯算法的过程,可以抽象为树形结构,因此解题的时候,可以通过画树形结构辅助做题。
回溯算法三部曲模版:
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}