回溯算法专题
回溯函数通常可以看做是递归函数,并且形象化为树的结构,回溯法解决的都是在集合中递归查找子集,集合的大小就构成了树的宽度,递归的深度就构成了树的深度。
模板:
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
思路:首先判断终止条件,当path路径中的长度与k相同则表示一定到达叶子结点,之后套用模板秒杀
代码:
class Solution {
private:
vector<vector<int>> res;
vector<int> path;
void backtracing(int n, int k,int startidx)
{
if(path.size() == k)
{
res.push_back(path);
return;
}
for(int i = startidx;i <= n;i++)
{
path.push_back(i);
backtracing(n,k,i+1);//i从取1到取4,中间每一次都回溯到叶子结点
path.pop_back();
}
}
public:
vector<vector<int>> combine(int n, int k) {
backtracing(n,k,1);
return res;
}
};
思路:大体思路同上,主要是需要提前获得组合的和以及判断条件需要更改
代码:
class Solution {
private:
vector<vector<int>> res;
vector<int> path;
void backtracing(int k, int n,int startidx)
{
int sum = 0;
for(int j = 0; j < path.size();j ++) sum += path[j];
if(path.size() == k && sum == n)
{
res.push_back(path);
return;
}
for(int i = startidx;i <= 9;i++)
{
path.push_back(i);
backtracing(k,n,i+1);
path.pop_back();
}
}
public:
vector<vector<int>> combinationSum3(int k, int n) {
backtracing(k,n,1);
return res;
}
};
思路:重点在于构造一个和电话上面的号码一一对应的映射表,因为如果直接对字符串digits进行遍历,获得的是字符,无法与数字结合。然后通过设置的index字符表索引来获得对应的号码的字符串,然后就是套模板。需要注意index在所有回溯算法中的作用基本都是用来从第一个组合尝试获得所有组合,利用递归和回溯。
代码:
class Solution {
private:
vector<string> res;
string path;
vector<string> digitToLetters = {
"", // 0
"", // 1
"abc", // 2
"def", // 3
"ghi", // 4
"jkl", // 5
"mno", // 6
"pqrs", // 7
"tuv", // 8
"wxyz" // 9
};
void backtracing(string digits,int index)//index表示当前使用电话字符表的索引位置
{
if(path.size() == digits.size())//终止条件是path路径等于字符串的长度,
{
res.push_back(path);
return;
}
//获取当前数字对应的字符串
int digit = digits[index] - '0';
string letters = digitToLetters[digit];//将所有输入的电话按钮对应的字符串存储起来
for(char str : letters)
{
path.push_back(str);
backtracing(digits,index + 1);
path.pop_back();
}
}
public:
vector<string> letterCombinations(string digits) {
if(digits.size() == 0) return {};
backtracing(digits,0);//index设为0,尝试所有组合
return res;
}
};