因为前几天课题组要开会,一直在搭实验,就没来得及打卡。现在攒了一堆ww,慢慢补了
回溯算法
代码随想录链接:代码随想录 (programmercarl.com)
回溯法也可以叫做回溯搜索法,它是一种搜索的方式。
只要有递归就会有回溯。
回溯的本质是穷举。
回溯可以解决:
- 组合问题:N个数里面按一定规则找出k个数的集合
- 切割问题:一个字符串按一定规则有几种切割方式
- 子集问题:一个N个数的集合里有多少符合条件的子集
- 排列问题:N个数按一定规则全排列,有几种排列方式(组合无序,排列有序)
- 棋盘问题:N皇后,解数独等等
回溯法解决的问题都可以抽象为树形结构,集合的大小就构成了树的宽度(for),递归的深度就构成了树的深度。
回溯模板:
1. 参数和返回值:返回值通常为void,参数较多,边写边添加
2. 终止条件:
if (终止条件) {
存放结果;
return;
}
3. 单层搜索逻辑:
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
77.组合
代码随想录链接:代码随想录 (programmercarl.com)
class Solution {
public:
vector<int> path;
vector<vector<int>> result;
void backtracking(int n, int k, int startindex){
if(path.size()==k){
result.push_back(path);
return;
}
for(int i=startindex;i<=n;i++){
path.push_back(i);
backtracking(n,k,i+1);
path.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
backtracking(n,k,1);
return result;
}
};
剪枝优化:
-
已经选择的元素个数:path.size();
-
还需要的元素个数为: k - path.size();
-
在集合n中至多要从该起始位置 : n - (k - path.size()) + 1,开始遍历
n=4,k=2的时候,第一层循环不必要遍历从4开始的情况。
优化的代码:
class Solution {
public:
vector<int> path;
vector<vector<int>> result;
void backtracking(int n, int k, int startindex){
if(path.size()==k){
result.push_back(path);
return;
}
for(int i=startindex;i<=n-(k-path.size())+1;i++){
path.push_back(i);
backtracking(n,k,i+1);
path.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
backtracking(n,k,1);
return result;
}
};
216.组合总和III
代码随想录链接:代码随想录 (programmercarl.com)
读题没读仔细,取数的范围是1-9。
class Solution {
public:
vector<int> path;
vector<vector<int>> result;
void backtracking(int k, int n, int startindex){
if(n<0){
return;
}
if(n==0&&path.size()==k){
result.push_back(path);
return;
}
for(int i=startindex;i<=9-(k-path.size())+1;i++){
path.push_back(i);
n-=i;
backtracking(k,n,i+1);
n+=i;
path.pop_back();
}
}
vector<vector<int>> combinationSum3(int k, int n) {
backtracking(k,n,1);
return result;
}
};
17.电话号码的字母组合
代码随想录链接:代码随想录 (programmercarl.com)
class Solution {
public:
const string letterMap[10] = {
"", // 0
"", // 1
"abc", // 2
"def", // 3
"ghi", // 4
"jkl", // 5
"mno", // 6
"pqrs", // 7
"tuv", // 8
"wxyz", // 9
};
vector<string> result;
string s;
void backtracing(const string& digits, int index){
if(index==digits.size()){
result.push_back(s);
return;
}
int digit=digits[index]-'0';
string letter=letterMap[digit];
for(int i=0;i<letter.size();i++){
s.push_back(letter[i]);
backtracing(digits,index+1);
s.pop_back();
}
}
vector<string> letterCombinations(string digits) {
if(digits.size()==0){
return result;
}
backtracing(digits,0);
return result;
}
};
总结
今天看了回溯算法的理论基础。写了三个题,主要思想是for循环处理宽度,递归处理深度,递归和回溯又是成对出现的。在必要的时候可以进行剪枝。