39. 组合总和
题目链接/文章讲解: 代码随想录视频讲解:带你学透回溯算法-组合总和(对应「leetcode」力扣题目:39.组合总和)| 回溯法精讲!_哔哩哔哩_bilibili
解题思路
这里和组合不同的是元素可以重复选取,其实也就是注意startindex的位置就可以,深度控制是由k的值来进行的
剪枝操作
剪枝一般都是在for循环上做操作,因为多了一些分支
而这题,我们只需要将数组排序后,例如235,和为4,当2+3已经大于4了,就没必要去遍历5了,因此在for循环中多加个判断条件即可
class Solution {
private:
vector<int> path;
vector<vector<int>> result;
int sum =0;
public:
void backtracking(vector<int>& candidates, int sum , int target , int startIndex)
{
if(sum == target)
{
result.push_back(path);
return;
}
for(int i = startIndex ; i<candidates.size()&& sum + candidates[i] <=target;i++) //将数组排序后,如果这个数和另一个数相加已经大于taget了,那么后面的数也就没有必要继续遍历了
{
sum+=candidates[i];
path.push_back(candidates[i]);
backtracking(candidates,sum,target,i);
path.pop_back();
sum-=candidates[i];
}
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
//对数组进行排序,方便后面的剪枝操作
sort(candidates.begin(),candidates.end());
backtracking(candidates,sum,target,0);
return result;
}
};
40.组合总和II
题目链接/文章讲解: 代码随想录视频讲解:回溯算法中的去重,树层去重树枝去重,你弄清楚了没?| LeetCode:40.组合总和II_哔哩哔哩_bilibili
解题思路
这道题中是有重复元素的,去重我们考虑,树枝去重(深)和树层去重(宽)
这题的去重逻辑,其实和三数之和很像,我们排完序后,在树枝上,我们取重复元素是可以的,但是在树层,在遇到相同的,也就是已经走过的路了,那就不要再走了,直接剪枝就可以了,但是我们如何去判断这是在树层上,还是在树枝上?我们只需要用uesd数组,当i-1的位置为0,说明在树层上,i-1的位置为1,说明在树枝上。
后面基本回溯的去重写法都是这个!!!非常重要
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
public:
void backtracking(vector<int>& candidates, int target, int sum , int startIndex , vector<int>& used)
{
if(sum == target)
{
result.push_back(path);
return;
}
for(int i= startIndex; i<candidates.size() && sum + candidates[i] <= target; i++)
{
if(i>0 &&candidates[i]== candidates[i-1] && used[i-1]==0) continue; //如果前一个used为1的话,表明在树枝上
path.push_back(candidates[i]);
sum += candidates[i];
used[i] = 1 ;
backtracking(candidates,target,sum, i+1,used);
used[i] = 0 ;
sum -= candidates[i];
path.pop_back();
}
}
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
vector<int> used(candidates.size(),0);
sort(candidates.begin(),candidates.end());
backtracking(candidates,target,0,0 , used);
return result;
}
};
131.分割回文串
代码随想录视频讲解:带你学透回溯算法-分割回文串(对应力扣题目:131.分割回文串)| 回溯法精讲!_哔哩哔哩_bilibili
解题思路
本题主要要理解的是如何画树形结构
其本质和组合问题是一样的,从图上说明更明确
这里我们进行切割的话,是使用startIndex来进行标记的,而在横向的切割中,是用i去寻找下一个切割点的,因此我们截取字符串也是从[stratIndex,i]去截取的
class Solution {
private:
vector<string> path;
vector<vector<string>> result;
bool isHuiWenStr(const string& s, int a, int b)
{
//利用双指针来判断回文串
for(int i=a,j=b;i<=j;i++,j--)
{
if(s[i]!=s[j])
return false;
}
return true;
}
void backtracking(const string& s, int startIndex) //这里是用startIndex来分割字符串的,本质和组合问题是一样的
{
//终止条件,因为是用startindex来分割的,那么当startindex的位置已经位于字符串最后说明分割完了
if(startIndex >= s.size())
{
result.push_back(path);
return;
}
//单层处理逻辑
for(int i= startIndex; i< s.size();i++)
{
if(isHuiWenStr(s,startIndex,i))
{
string str = s.substr(startIndex,i-startIndex+1); //截取字符串函数,变量分别为索引和个数
path.push_back(str);
}
else
continue;
backtracking(s,i+1);
path.pop_back();
}
}
public:
vector<vector<string>> partition(string s) {
backtracking(s,0);
return result;
}
};
收获
继续加油