给定一个无重复元素的数组 candidates
和一个目标数 target
,找出 candidates
中所有可以使数字和为 target
的组合。
candidates
中的数字可以无限制重复被选取。
说明:
- 所有数字(包括
target
)都是正整数。 - 解集不能包含重复的组合。
示例 1:
输入: candidates =[2,3,6,7],
target =7
, 所求解集为: [ [7], [2,2,3] ]
示例 2:
输入: candidates = [2,3,5],
target = 8,
所求解集为:
[
[2,2,2,2],
[2,3,3],
[3,5]
]
【思路】回溯法的经典实用场景,注意这道题要求每个数可以无限选择,所以每一层都有n种选择,在递归向下层的时候不用 level+1,而是直接传level 就可以,直到sum == target 或者 sum > target ,再返回上层。注意这里分开写是因为等于满足条件,可以加到最终解中,而大于不满足条件直接返回。代码如下:
class Solution {
public:
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
vector<vector<int> > res;
vector<int> out;
int sum = 0;
helper(target,sum,0,out,candidates,res);
return res;
}
void helper(int target,int sum,int level,vector<int>& out,vector<int>& candidates,vector<vector<int>>& res){
if(sum > target)
return;
if(target == sum){
res.push_back(out);
return;
}
for(int i = level; i < candidates.size(); ++i){//for循环中的i横向增加是在自己的兄弟节点中遍历
out.push_back(candidates[i]);
helper(target,sum + candidates[i],i,out,candidates,res);//每次递归是往下走一层
out.pop_back();
}
}
};
上面的写法我个人感觉比较好理解,以下是上面代码的简化版:
class Solution {
public:
vector<vector<int>> ans;
vector<int> t;
void dfs(int level, int target, const vector<int>& v)
{
if(target < 0 || level == v.size())
return;
if(target == 0){
ans.push_back(t);
return;
}
if(target >= v[level]){
t.push_back(v[level]);
dfs(level, target - v[level], v);
t.pop_back();
}
dfs(level + 1, target, v);
return;
}
vector<vector<int>> combinationSum(vector<int>& v, int target) {
dfs(0, target, v);
return ans;
}
};