思路:需要一个参数startIndex来指定for循环的起始位置,注意题目允许同一个数字无限制地被使用, 因此递归传参时startIndex应该传i,而不是i+1。
代码:
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
//sum用来统计path数组里的元素和,startIndex来控制for循环的起始位置
//如果是一个集合来求组合的话,就需要startIndex;如果是多个集合取组合,各个集合之间相互不影响,那么就不用startIndex
void backtracking(vector<int>& nums, int sum, int target, int startIndex){
if(sum > target) return;
if(sum == target){
result.push_back(path);
return;
}
for(int i = startIndex; i < nums.size(); i++){
path.push_back(nums[i]);
sum += nums[i];
backtracking(nums, sum, target, i);//题目允许同一个数字可以无限制地重复被选取,所以for循环的起始位置可以从i开始,不用i+1表示可以重复读取当前数值
sum -= nums[i];
path.pop_back();
}
return;
}
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
result.clear();
path.clear();
if(candidates.size() == 0) return result;
if(candidates.size() == 1 && candidates[0] > target) return result;
backtracking(candidates, 0, target, 0);
return result;
}
};
思路: 因为集合中元素有重复的,所以在组合的过程中需要对元素进行去重。
首先对集合进行排序,让重复的元素紧挨着一起。之后在for循环起始处对元素进行去重。
代码:
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int target, int sum, int startIndex){
if(sum > target) return;
if(sum == target){
result.push_back(path);
return;
}
for(int i = startIndex; i < nums.size(); i++){
//对同一树层的元素进行去重,因为集合包含重复元素,如果使用相同元素解集可能会重复
if(i > startIndex && nums[i] == nums[i - 1]){
continue;
}
path.push_back(nums[i]);
sum += nums[i];
backtracking(nums, target, sum, i + 1);//要求每个数字只能使用一次,所以下一个for循环起始位置startIndex要从i + 1位置开始
sum -= nums[i];
path.pop_back();
}
}
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
path.clear();
result.clear();
sort(candidates.begin(), candidates.end());//首先对集合排序,让重复的元素紧挨着一起
backtracking(candidates, target, 0, 0);
return result;
}
};