文章目录
- 39.组合总和
- 思路
- 伪代码实现
- 剪枝优化
- CPP代码
- 普通版本
- 剪枝版本
- 40.组合总和II
- 思路
- 伪代码
- CPP代码
- 不采用used数组的去重方法
- 组合问题小总结
- 131.分割回文串
- 思路
- 伪代码
- CPP代码
39.组合总和
力扣题目链接
文章讲解:39.组合总和
视频讲解:Leetcode:39. 组合总和讲解
状态:强烈建议回看本篇文章内容组合总和III、电话号码的字母组合。基本就是沿用了这两门道题的主要思想:单层递归中算加法、index控制循环的深度。
思路
伪代码实现
-
递归函数参数
- 两个全局变量:
result
存放结果集,数组path
存放符合条件的结果 - 定义
int
型的sum
变量来统计单一结果path
里的总和,其实这个sum
也可以不用,用target
做相应的减法就可以了,最后如何target==0
就说明找到符合的结果了,但为了代码逻辑清晰,依然用了sum
- 使用startIndex控制for循环的起始位置。
vector<vector<int>> result; vector<int> path; void backtracking(vector<int>& candidates, int target, int sum, int startIndex)
- 两个全局变量:
-
递归终止条件
从叶子结点可以看出,终止就是这两种情况,因为我们递归的纵向深度是动态的、不确定的。
if (sum > target){return ;
}
if (sum == target){result.push_back(path);return ;
}
- 单层搜索的逻辑
单层for循环依然从startIndex开始,搜索candidates集合。但是一定要注意,从树形结构可以看出,我们本题元素是可以重复选取的,那么如何实现呢?最关键的代码就是:
backtracking(candidates, target, sum, i);
关键点:不用i+1
了,表示可以重复读取当前的数。
for (int i = startIndex; i < candidates.size(); i++){sum += candidates[i];path.push_back(candidates[i]);backtracking(candidates, target, sum, i);sum -= candidates[i];path.pop_back();
}
剪枝优化
本题中,对于sum已经大于target的情况,其实是依然进入了下一层递归,只是下一层递归结束判断的时候,会判断sum > target的话就返回。其实如果已经知道下一层sum会大于target的话,完全没有必要进入下一层递归,那么如果能够保证candidates数组是递增的就好了。
对总集合排序之后,如果下一层的sum(就是本层的 sum + candidates[i])已经大于target,就可以结束本轮for循环的遍历。
所以我们对for循环的剪枝如下:
for (int i = startIndex; i < candidates.size() && sum + candidates[i] <= target; i++)
CPP代码
普通版本
class Solution {
private:vector<vector<int>> result;vector<int> path;void backtracking(vector<int>& candidates, int target, int sum, int startIndex) {if (sum > target) {return;}if (sum == target) {result.push_back(path);return;}for (int i = startIndex; i < candidates.size(); i++) {sum += candidates[i];path.push_back(candidates[i]);backtracking(candidates, target, sum, i); // 不用i+1了,表示可以重复读取当前的数sum -= candidates[i];path.pop_back();}}
public:vector<vector<int>> combinationSum(vector<int>& candidates, int target) {result.clear();path.clear();backtracking(candidates, target, 0, 0);return result;}
};
剪枝版本
class Solution {
private:vector<vector<int>> result;vector<int> path;void backtracking(vector<int>& candidates, int target, int sum, int startIndex) {if (sum == target) {result.push_back(path);return;}// 如果 sum + candidates[i] > target 就终止遍历for (int i = startIndex; i < candidates.size() && sum + candidates[i] <= target; i++) {sum += candidates[i];path.push_back(candidates[i]);backtracking(candidates, target, sum, i);sum -= candidates[i];path.pop_back();}}
public:vector<vector<int>> combinationSum(vector<int>& candidates, int target) {result.clear();path.clear();sort(candidates.begin(), candidates.end()); // 需要排序backtracking(candidates, target, 0, 0);return result;}
};
40.组合总和II
力扣题目链接
文章讲解:40.组合总和II
视频讲解:回溯算法中的去重,树层去重树枝去重,你弄清楚了没?| LeetCode:40.组合总和II
状态:本题最重要的——集合有重复元素,但是不能有重复的组合。首先要注意的是set、map驱虫是很容易潮湿的,只能在搜索的过程中就去掉重复组合。
这里点名两个重要的概念:树枝去重、树层去重
思路
这里我们来理解本题中的去重。
都知道组合问题可以抽象为树形结构,那么“使用过”在这个树形结构上是有两个维度的,一个维度是同一树枝上使用过,一个维度是同一树层上使用过。没有理解这两个层面上的“使用过” 是造成大家没有彻底理解去重的根本原因。
首先,同一树层使用会造成重复,同一树枝使用也会造成重复。而题目要求是,组合内可以重复,但两个组合不能相同。我们结合图片来看:
总而言之,我们要去重的是同一树层上的“使用过”,同一树枝上的都是一个组合里的元素,不用去重。
强调一下,树层去重的话,需要对数组排序!
伪代码
- 递归函数参数:与39.组合总和套路相同,此题还需要加一个
bool
型数组used
,用来记录同一树枝上的元素是否使用过,这个集合去重的重任就是used
来完成的。
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& candidates, int target, int sum, int startIndex, vector<bool>& used)
- 终止递归条件
if (sum > target){ //如果执行剪枝操作,该步骤可以省略return ;
}
if (sum == target){result.push_back(path);return ;
}
-
单层搜索的逻辑:
- 本流程中最终要的就是去重“同一树层上的元素被使用过”。首先我们明确,同树层,我们用bool used来标识我们取的哪一个数,然后同树枝bool used来标识被取过的数。如下图所示:
- used[i-1] == true, 说明同一树枝candidates[i-1]被使用过,说明现在应该进入下一层递归,取一下个数。。
- Used[i-1] == false, 说明同一树层candidates[i-1]被使用过,表示了当前取的candidates[i]是从candidates[i-1]回溯过来的,是树层上的。
- 本流程中最终要的就是去重“同一树层上的元素被使用过”。首先我们明确,同树层,我们用bool used来标识我们取的哪一个数,然后同树枝bool used来标识被取过的数。如下图所示:
for (int i = startIndex; i < candidates.size() && sum + candidates[i] <= traget; i++){// used[i - 1] == true,说明同一树枝candidates[i - 1]使用过// used[i - 1] == false,说明同一树层candidates[i - 1]使用过// 要对同一树层使用过的元素进行跳过if (i > 0 && candidate[i] == candidates[i-1] && used[i-1] == false){continue;}sum += candidates[i];path.push_back(candidates[i]);used[i] = true;backtracking(candidates, target, sum, i + 1, used);used[i] = false;sum -= candidates[i];path.pop_back();
}
说了这么多,其实used最主要的作用就是当前的遍历到底是在树枝上,还是在树层上。
CPP代码
class Solution {
private:vector<vector<int>> result;vector<int> path;void backtracking(vector<int>& candidates, int target, int sum, int startIndex, vector<bool>& used) {if (sum == target) {result.push_back(path);return;}for (int i = startIndex; i < candidates.size() && sum + candidates[i] <= target; i++) {// used[i - 1] == true,说明同一树枝candidates[i - 1]使用过// used[i - 1] == false,说明同一树层candidates[i - 1]使用过// 要对同一树层使用过的元素进行跳过if (i > 0 && candidates[i] == candidates[i - 1] && used[i - 1] == false) {continue;}sum += candidates[i];path.push_back(candidates[i]);used[i] = true;backtracking(candidates, target, sum, i + 1, used); // 和39.组合总和的区别1,这里是i+1,每个数字在每个组合中只能使用一次used[i] = false;sum -= candidates[i];path.pop_back();}}public:vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {vector<bool> used(candidates.size(), false);path.clear();result.clear();// 首先把给candidates排序,让其相同的元素都挨在一起。sort(candidates.begin(), candidates.end());backtracking(candidates, target, 0, 0, used);return result;}
};
不采用used数组的去重方法
class Solution {
private:vector<vector<int>> result;vector<int> path;void backtracking(vector<int>& candidates, int target, int sum, int startIndex) {if (sum == target) {result.push_back(path);return;}for (int i = startIndex; i < candidates.size() && sum + candidates[i] <= target; i++) {// 要对同一树层使用过的元素进行跳过//startIndex代表这一层中首次尝试的元素,//只有当'i>index'时,才说明我们在同一树层中考虑是否使用重复元素if (i > startIndex && candidates[i] == candidates[i - 1]) {continue;}sum += candidates[i];path.push_back(candidates[i]);backtracking(candidates, target, sum, i + 1); // 和39.组合总和的区别1,这里是i+1,每个数字在每个组合中只能使用一次sum -= candidates[i];path.pop_back();}}public:vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {path.clear();result.clear();// 首先把给candidates排序,让其相同的元素都挨在一起。sort(candidates.begin(), candidates.end());backtracking(candidates, target, 0, 0);return result;}
};
组合问题小总结
- 什么时候要用startIndex来控制for循环,什么时候不用呢?
如果是一个集合来求组合的话,就需要用startIndex,例如:77.组合,216.组合总和III
如果是多个集合取组合,各个集合之间相互不影响,那么就用startIndex,例如:17.电话号码的字母组合。
131.分割回文串
力扣题目链接
文章讲解:131.分割回文串
视频讲解:131.分割回文串
状态:毫无想法,树形结构都画不出来
本题最主要是两个关键问题:
- 切割问题,有不同的切割方式
- 判断回文
思路
我们来分析一下切割,其实切割问题类似组合问题。
例如对于字符串abcdef:
- 组合问题:选取一个a之后,在bcdef中再去选取第二个,选取b之后在cdef中再选取第三个…。
- 切割问题:切割一个a之后,在bcdef中再去切割第二段,切割b之后在cdef中再切割第三段…。
现在我们再来回答一下上文中的问题:
- 我们树形结构的层第一次切割,截取第一个字母,第二次切割截取前两个字母
- 我们对于树形结构的深就是每次判断一下是否为回文即可。
树形结构如图:递归用来总想遍历,for循环用来横向遍历。
伪代码
- 递归函数参数
全局变量数组path存放切割后回文的子串,二维数组result存放结果集。 (这两个参数可以放到函数参数里)
本题递归函数参数还需要startIndex,因为切割过的地方,不能重复切割,和组合问题也是保持一致的。
vector<vector<string>> result;
vector<string> path; // 放已经回文的子串
void backtracking (const string& s, int startIndex) {
- 递归终止条件
红色线条为切割线,每次切割完都表示找到了一种切割方法,都进行一次回文子串的判断,
void backtracking(const string& s, int startIndex){//如果起始位置已经大于s的大小,说明已经找到了一组分割方案if (startIndex >= s.size()){result.push_back(path);return;}
}
- 单层搜索的逻辑
递归循环中如何截取子串?
for (int i = startIndex; i < s.size(); i++){if (isPalindrom(s, startIndex, i)){//是回文子串//获取[startIndex, i]在s中的子串string str = s.substr(startIndex, i - startIndex + 1);path.push_back(str);}else{continue;//不是回文就直接跳过}backtracking(s, i + 1); //寻找i+1为起始位置的子串path.pop_back(); //回溯过程,染出本次已经调价的子串
}
注意切割过的位置,不能重复切割,所以,backtracking(s, i + 1); 传入下一层的起始位置为i + 1。
判断回文子串?
最后我们看一下回文子串要如何判断了,判断一个字符串是否是回文。
可以使用双指针法,一个指针从前向后,一个指针从后向前,如果前后指针所指向的元素是相等的,就是回文字符串了。
bool isPalindrome(const string& s, int start, int end){for (int i = start, j = end; i < j; i++, j--){if (s[i] != s[j]){return false;}}return true;
}
CPP代码
class Solution {
private:vector<vector<string>> result;vector<string> path; // 放已经回文的子串void backtracking (const string& s, int startIndex) {// 如果起始位置已经大于s的大小,说明已经找到了一组分割方案了if (startIndex >= s.size()) {result.push_back(path);return;}for (int i = startIndex; i < s.size(); i++) {if (isPalindrome(s, startIndex, i)) { // 是回文子串// 获取[startIndex,i]在s中的子串string str = s.substr(startIndex, i - startIndex + 1);path.push_back(str);} else { // 不是回文,跳过continue;}backtracking(s, i + 1); // 寻找i+1为起始位置的子串path.pop_back(); // 回溯过程,弹出本次已经添加的子串}}bool isPalindrome(const string& s, int start, int end) {for (int i = start, j = end; i < j; i++, j--) {if (s[i] != s[j]) {return false;}}return true;}
public:vector<vector<string>> partition(string s) {result.clear();path.clear();backtracking(s, 0);return result;}
};