今天是回溯算法学习的第一天,主要的学习内容包括:回溯算法的基本原理以及用于解决的问题类型,在单集合中进行组合问题,在多集合中进行组合问题。
回溯算法理论基础
回溯算法是一种搜索的方法,一般和递归是一起使用的。回溯算法的性能并不好,其本质是穷举,一般用剪枝进行优化,用于解决暴力搜索问题。常见的题型有下面几种:
1.组合问题:N个数字中按一定规律找出K个数的组合
2.切割问题:一个字符串按某种规则进行切割
3.子集问题:一个N个数的集合有多少符合条件的子集
4.排列问题:N个数按一定规则排列,有哪些排列结果
5.棋盘问题:N皇后,解数独等
在上面的题型中,需要注意的是,组合和排列是不同的。组合不要求顺序,排列则对顺序是有要求的。
在解决回溯法的问题时,可以将回溯法的问题抽象为树形结构,集合大小决定了树的宽度,递归深度决定了树的深度。
回溯法也是有对应的模版的。首先对于函数返回值,一般是void,函数名习惯为backtracking,参数则根据需要进行选择。
void backtracking(参数)
和递归一样,首先需要确定终止条件。一般写法如下:
if (终止条件) {
作出处理;
return;
}
接下来是递归遍历的过程,每一层对应的就是一层树的遍历,使用for循环对这棵树的一层的每个结点进行遍历。需要注意的是,处理完之后需要进行回溯。
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
完整的代码模版如下所示:
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
77. 组合
这个题就是在N个数中找到选取K个数进行组合。首先是参数,我们需要N和K,还需要一个值记录当前层取的是哪一个数。我们用startindex进行记录。
终止条件就比较简单了,当vec数组内的元素个数达到K个时,就放入result中并进行返回。
对于单层的处理逻辑,我们需要处理从startindex开始到末尾的所有数字,分别放入vec,进行下一层递归,然后进行回溯。具体完整代码的实现如下所示:
class Solution {
public:
vector<int> vec;
vector<vector<int>> result;
void backtracking(int n,int k,int startindex)
{
if(vec.size()==k)
{
result.push_back(vec);
return;
}
for(int i=startindex;i<=n;i++)
{
vec.push_back(i);
backtracking(n,k,i+1);
vec.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
result.clear();
vec.clear();
backtracking(n,k,1);
return result;
}
};
这个题还可以进行一些剪枝操作。主要是在单层遍历的过程中,如果把剩余的数遍历完都没有K个,那么就可以对后面的遍历进行剪枝了。剪枝后的代码实现如下:
class Solution {
public:
vector<int> vec;
vector<vector<int>> result;
void backtracking(int n,int k,int startindex)
{
if(vec.size()==k)
{
result.push_back(vec);
return;
}
for(int i=startindex;i<=n-(k-vec.size())+1;i++)//具体的剪枝部分在这里
{
vec.push_back(i);
backtracking(n,k,i+1);
vec.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
result.clear();
vec.clear();
backtracking(n,k,1);
return result;
}
};
216.组合总和III
题目链接:216. 组合总和 III - 力扣(LeetCode)
这个题目和上面那题是相似的,无非是找到K个数后需要判断求和是否是满足n的。其次在剪枝上还可以进行优化,如果当求和数没有K个但是和已经大于n,那就可以直接返回了。具体代码实现如下:
class Solution {
public:
vector<int> vec;
vector<vector<int>> result;
void backtracking(int k,int n,int startindex,int sum)
{
if(sum>n) return;
if(vec.size()==k)
{
if(sum==n) result.push_back(vec);
return;
}
for(int i=startindex;i<10-(k-vec.size())+1;i++)
{
vec.push_back(i);
backtracking(k,n,i+1,sum+i);
vec.pop_back();
}
}
vector<vector<int>> combinationSum3(int k, int n) {
backtracking(k,n,1,0);
return result;
}
};
17.电话号码的字母组合
题目链接:17. 电话号码的字母组合 - 力扣(LeetCode)
这个题目的意思是给定一个数字串。每一个数字可能对应三到四个字母,需要将这些字母的所有组合结果进行输出。这个题目跟前面的组合是不太一样的,主要的不同在于递归回溯逻辑上。使用树的思想进行解题,这棵树的每一层是一个数字对应的字母的各种情况。所以只需要记录递归的深度,也就是遍历到了数字串的第几个数字即可。具体代码的实现如下:
class Solution {
public:
string num[10]={"","","abc","def","ghi","jkl","mno","pqrs","tuv","wxyz"};
string s;
vector<string> result;
void backtracking(string digits,int index)
{
if(index==digits.size())
{
result.push_back(s);
return;
}
int size=digits[index]-'0';
string letter=num[size];
for(int i=0;i<letter.size();i++)
{
s.push_back(letter[i]);
backtracking(digits,index+1);
s.pop_back();
}
}
vector<string> letterCombinations(string digits) {
if(digits=="") return result;
backtracking(digits,0);
return result;
}
};