LeetCode39. 组合总和
该题与之前的题相比涉及两个重要思路:
1、去重:遍历当前元素时,重复回溯,可以再包含当前元素,但是不能再包含上一个元素,否则就会有重复,因此这里回溯参数选取的是i,而不是i+1;
2、剪枝:因为数组不是有序的,如果按以前的方法剪枝,则会少掉很多结果,如果不排序,那么则需要在终止条件上加入return的逻辑,如果排序,则就可以在for循环里面剪枝。
代码如下:最坏时间复杂度(n*2^n);空间复杂度O(targer)
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
int sum = 0;
void backtracking(vector<int>& candidates,int target,int startIndex){
if(sum==target){
result.push_back(path);
return;
}
for(int i=startIndex;i<candidates.size();i++){
sum += candidates[i];
if(sum>target){
sum -= candidates[i];
return;
}
path.push_back(candidates[i]);
backtracking(candidates,target,i);
path.pop_back();
sum -= candidates[i];
}
return;
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
sort(candidates.begin(),candidates.end());
backtracking(candidates,target,0);
return result;
}
};
LeetCode40.组合总和II
这道题的精髓在于去重操作,因为如果相邻两个数相同,那么取完第一个数再取后面的集合区间,包含了取完第二个数再取后面的集合区间,因此取完第二个数再去后面的数的集合区间一定是重复的,需要去掉。
那么这里面的去重操作,就是当在for循环里,不是第一层循环的时候,取得数得值与前一个数相等,那么就可以continue,利用startIndex来判断,就用不到used数组了。代码如下:
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
int sum = 0;
void backtracking(vector<int>& candidates,int target,int startIndex){
if(sum==target){
result.push_back(path);
return;
}
for(int i=startIndex;i<candidates.size();i++){
if(i>startIndex && candidates[i]==candidates[i-1]){
continue;
}
sum += candidates[i];
if(sum>target){
sum -= candidates[i];
return;
}
path.push_back(candidates[i]);
backtracking(candidates,target,i+1);
path.pop_back();
sum -= candidates[i];
}
return;
}
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
sort(candidates.begin(),candidates.end());
backtracking(candidates,target,0);
return result;
}
};
LeetCode131.分割回文串
这里面的难点在于:
- 切割问题可以抽象为组合问题
- 如何模拟那些切割线
- 切割问题中递归如何终止
- 在递归循环中如何截取子串
- 如何判断回文
主要是想清楚切割线如何转化为组合问题后,收取子字符串判断回文即可,代码如下:(未做剪枝操作)
class Solution {
public:
vector<vector<string>> result;
vector<string> path;
void backtracking(string s,int startIndex){
if(startIndex==s.size()){
result.push_back(path);
return;
}
for(int i=startIndex;i<s.size();i++){
if(!isPalindrome(s,startIndex,i)) continue;
else{
string str = s.substr(startIndex,i-startIndex+1);
path.push_back(str);
}
backtracking(s,i+1);
path.pop_back();
}
return;
}
bool isPalindrome(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;
}
vector<vector<string>> partition(string s) {
backtracking(s,0);
return result;
}
};
剪枝操作的思路就是如果判断bcd不是回文,那么就不用重新判断abcde是不是回文了,因此在进行回溯之前维护一个表格记录所有的子字符串是否是回文,然后在回溯的过程中直接查表就可以得到,不需要每一次回溯都进行一次双指针操作。
当然,这里面建立这样一个表格就足以作为力扣的另外一道有难度的题目,思路也是比较复杂,这里直接贴一下卡哥的代码如下:
class Solution {
private:
vector<vector<string>> result;
vector<string> path; // 放已经回文的子串
vector<vector<bool>> isPalindrome; // 放事先计算好的是否回文子串的结果
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[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(); // 回溯过程,弹出本次已经添加的子串
}
}
void computePalindrome(const string& s) {
// isPalindrome[i][j] 代表 s[i:j](双边包括)是否是回文字串
isPalindrome.resize(s.size(), vector<bool>(s.size(), false)); // 根据字符串s, 刷新布尔矩阵的大小
for (int i = s.size() - 1; i >= 0; i--) {
// 需要倒序计算, 保证在i行时, i+1行已经计算好了
for (int j = i; j < s.size(); j++) {
if (j == i) {isPalindrome[i][j] = true;}
else if (j - i == 1) {isPalindrome[i][j] = (s[i] == s[j]);}
else {isPalindrome[i][j] = (s[i] == s[j] && isPalindrome[i+1][j-1]);}
}
}
}
public:
vector<vector<string>> partition(string s) {
result.clear();
path.clear();
computePalindrome(s);
backtracking(s, 0);
return result;
}
};