给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合,candidates 中的数字可以无限制重复被选取,示例如下:
数组 candidates 中不包含0,如果有0要考虑无限循环的情况;数组中不包含重复元素,若包含重复元素要考虑去重的问题。
本题与组合问题的区别是不限定组合元素的数量,并且可以重复选取。
回溯三部曲:
(1)参数和返回值:返回值为void,参数根据后面代码确定。
(2)返回条件:返回条件需要根据组合的和来确定,组合的和大于目标值就返回。
(3)单层搜索逻辑:树的结构搜索,树的宽度为集合个数,深度由返回条件确定。
实现代码如下:
//组合求和
vector<int> vec;
vector<vector<int>> result;
int sum = 0;
void backtracking(vector<int>& candidates, int start, int& target) {
//确定终止条件
if(sum == target) {
result.push_back(vec);
return;
}
else if(sum > target) {
return;
}
//单层搜索的逻辑
for(int i = start; i<candidates.size(); i++) {
vec.push_back(candidates[i]);
sum += candidates[i];
backtracking(candidates, i, target); //递归
sum -= candidates[i]; //回溯
vec.pop_back();
}
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
vec.clear();
result.clear();
backtracking(candidates, 0, target);
return result;
}
4.增加剪枝操作
增加剪枝,如果加第 i 个元素已经大于目标值之后,就不再遍历后面的元素,但前提是需要对数组进行排序,实现代码如下:
//增加剪枝
vector<int> vec;
vector<vector<int>> result;
int sum = 0;
void backtracking(vector<int>& candidates, int start, int& target) {
//确定终止条件
if(sum == target) {
result.push_back(vec);
return;
}
else if(sum > target) {
return;
}
//单层搜索的逻辑 增加剪枝
//增加剪枝,如果加i已经大于目标值之后,就不再使用i+1
//前提是需要给数组进行排序
for(int i = start; i<candidates.size() && sum + candidates[i] <= target; i++) {
vec.push_back(candidates[i]);
sum += candidates[i];
sorted(vec.begin(), vec.end());
backtracking(candidates, i, target); //递归
sum -= candidates[i]; //回溯
vec.pop_back();
}
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
vec.clear();
result.clear();
backtracking(candidates, 0, target);
return result;
}
在求和问题中,排序之后再进行剪枝操作是常用的方法。