题目:
给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。
思路:
回溯三部曲:
1、递归函数参数
1.1、定义两个全局变量,二维数组result存放结果集,数组path存放符合条件的结果;
1.2、题目中给出的参数,集合candidates, 和目标值target;
1.3、定义了int型的sum变量来统计单一结果path里的总和;
1.4、还需要startIndex来控制for循环的起始位置。
部分代码如下:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& candidates, int target, int sum, int startIndex)
2、递归终止条件
本题搜索过程抽象成树形结构如下:
从叶子节点可以清晰看到,终止只有两种情况,sum大于target和sum等于target。
sum等于target的时候,需要收集结果,代码如下:
if (sum >= target) {
result.push_back(path);
return;
}
3、单层搜索逻辑
单层for循环依然是从startIndex开始,搜索candidates集合,本题元素为可重复选取。代码如下:
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(); // 回溯
}
整体代码如下:
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
void backtrack(vector<int>& nums,int start,int sum,int target)
{
// 返回条件
if(sum > target) return;
if(sum == target)
{
result.push_back(path);
return;
}
// 单层搜索逻辑
for(int i=start;i<nums.size();i++)
{
sum += nums[i];
path.push_back(nums[i]);
backtrack(nums,i,sum,target); // 不用i+1了,表示可以重复读取当前的数
sum -= nums[i]; // 回溯
path.pop_back(); // 回溯
}
}
public:
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
backtrack(candidates,0,0,target);
return result;
}
};