题目大意
给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的数字可以无限制重复被选取。
说明:
所有数字(包括 target)都是正整数。
解集不能包含重复的组合。
解题思路
设置一个递归函数dfs(i, target),表示从i位置开始,找到能出target的方案。当递归过程中的和等于target时,将该方案放入结果中。否则遍历到当前位置时,有两种选择:(1)将当前数字加入,由于可以重复选取,因此下一步仍然从当前位置开始递归;(2)不考虑当前位置,直接从下一个位置开始递归;
class Solution {
public:
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
if (candidates.empty())
return {};
vector<vector<int>> res;
//从0位置开始,当前和为0,目标和为target
dfs(res, candidates, {}, 0, 0, target);
return res;
}
void dfs(vector<vector<int>> & res, vector<int> & candidates, vector<int> ans, int index, int curSum, int target){
if (curSum == target){
res.push_back(ans);
return;
}
if (curSum > target)
return ;
if (index == candidates.size())
return ;
//考虑当前位置,因此将当前位置放入缓存数组中
ans.push_back(candidates[index]);
dfs(res, candidates, ans, index, curSum + candidates[index], target);
//上述情况考虑完毕后应弹出之前放入的值
ans.pop_back();
//情况2,不考虑当前位置,直接从下一个位置开始考虑
dfs(res, candidates, ans, index + 1, curSum, target);
}
};