题目描述
给定一个无重复元素的数组 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] ]
思路
回溯思路。因为题中说每个数可以重复选取,所以循环的索引为从当前值开始。如何去重呢,先排序即可。
代码
class Solution {
public:
vector<vector<int>> res;
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
vector<int> tmp;
sort(candidates.begin(), candidates.end());
dfs(candidates, target, tmp, 0);
return res;
}
void dfs(vector<int> &arr, int target, vector<int> tmp, int idx){
if (0 >= target){
if (0 == target) res.emplace_back(tmp);
return;
}
for (int i = idx; i < arr.size(); ++i){
if (i == 0 || (i > 0 && arr[i] != arr[i-1])){
tmp.emplace_back(arr[i]);
dfs(arr, target-arr[i], tmp, i);
tmp.pop_back();
}
}
}
};