题目
给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。
candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。
对于给定的输入,保证和为 target 的不同组合数少于 150 个。
解题思路
本题容易想到类似二叉树的某条路径和为target, 因此可使用回溯算法。
首先定义result和path来表示最终的组合结果,和单条满足条件的路径。因为是在同一数组中选取多次,则需要用start_index区分起始点。
其次定义backtracking(vector& candidates, int target, int sum, int startIndex)的形式和单层逻辑,递归调用需要设置结束条件,如果数据之和sum大于target值,则说明该条路径不可能等于target,return结束;如果数据之和sum等于target值,则说明该path符合要求,将其值保存到result里,再return结束;如果数据之和小于target,不需中断,继续遍历。
最后需要回溯,清除已经遍历的path,重新遍历新的path。
代码实现
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& candidates, int target, int sum, int startIndex) {
if (sum > target) {
return;
}
if (sum == target) {
result.push_back(path);
return;
}
for (int i = startIndex; i < candidates.size(); i++) {
sum += candidates[i];
path.push_back(candidates[i]);
backtracking(candidates, target, sum, i);
sum -= candidates[i];
path.pop_back();
}
}
public:
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
result.clear();
path.clear();
backtracking(candidates, target, 0, 0);
return result;
}
};