代码随想录--回溯部分
day 23 回溯部分第二天
文章目录
一、力扣39--组合总和
代码随想录题目链接:代码随想录
给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。
candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。
对于给定的输入,保证和为 target 的不同组合数少于 150 个。
其实和之前的组合总和一样,只是多了个可以重复选取,那么在回溯的时候startIndex不需要加一即可
代码如下:
cpp
class Solution {
public:
int sum = 0;
vector<int> path;
vector<vector<int>> result;
void backTracking(vector<int> candidates, int target, int start)
{
if(sum > target)return;
else if (sum == target)
{
result.push_back(path);
return;
}
for(int i = start; i < candidates.size(); i ++)
{
sum += candidates[i];
path.push_back(candidates[i]);
backTracking(candidates, target, i);
sum -= candidates[i];
path.pop_back();
}
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
backTracking(candidates, target, 0);
return result;
}
};
二、力扣40--组合总和Ⅱ
代码随想录题目链接:代码随想录
给定一个候选人编号的集合 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用 一次 。
其实就是多了个要求:答案去重,粗暴做肯定会超时
所以使用新的数组used来记录每个元素是否被使用过,如果使用过了就跳过
这点有些像力扣15三数之和的去重,应当比较candidates [i]和[i-1]是否一样,一样就把i跳过,因为这样跳过的是搜索过的元素
同时还要通过used来判断目前是同层还是不同层
如果是true,说明这个重复的值是上层递归的,那就可以取,如果为false,说明是同层递归,那再取就会重复了
代码如下:
cpp
class Solution {
public:
int sum = 0;
vector<int> path;
vector<vector<int>> result;
vector<bool> used;
void backTracking(vector<int> candidates, int target, int start)
{
if(sum > target)return;
else if (sum == target)
{
result.push_back(path);
return;
}
for(int i = start; i < candidates.size(); i ++)
{
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, i+1);
sum -= candidates[i];
path.pop_back();
used[i] = false;
}
}
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
used = vector<bool>(candidates.size(), false);
sort(candidates.begin(), candidates.end());
backTracking(candidates, target, 0);
return result;
}
};
还是挺抽象的,借助used来表示递归的层数,或许用栈能够更直白一些?
三、力扣--分割回文串
代码随想录题目链接:代码随想录
给你一个字符串 s,请你将 s 分割成一些子串,使每个子串都是 回文串 。返回 s 所有可能的分割方案。
照着图写就行
代码如下:
cpp
class Solution {
public:
vector<vector<string>> result;
vector<string> path;
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;
}
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))
{
string temp = s.substr(startIndex, i - startIndex + 1);
path.push_back(temp);
}
else continue;
backTracking(s, i + 1);
path.pop_back();
}
}
vector<vector<string>> partition(string s) {
backTracking(s, 0);
return result;
}
};
值得一讲的是在回溯前的处理,并不是每个遍历都值得接着向下回溯,实际上如果本次搜索并不是回文串,就可以结束了
因为结果中必须都是回文串,所以一直要搜索到回文串后,才有资格接着向下搜索。