给定一个无重复元素的正整数数组 candidates 和一个正整数 target ,找出 candidates 中所有可以使数字和为目标数 target 的唯一组合。
candidates 中的数字可以无限制重复被选取。如果至少一个所选数字数量不同,则两种组合是唯一的。
对于给定的输入,保证和为 target 的唯一组合数少于 150 个。
示例 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]]
示例 3:输入: candidates = [2], target = 1
输出: []
示例 4:输入: candidates = [1], target = 1
输出: [[1]]
示例 5:输入: candidates = [1], target = 2
输出: [[1,1]]
提示:
1 <= candidates.length <= 30
1 <= candidates[i] <= 200
candidate 中的每个元素都是独一无二的。
1 <= target <= 500
解题思路:
- 本题是一个关于递归的问题,递归问题关键是知道递归的终止条件,递归结束需要什么操作。
- 然后处理单层递归。
- 函数体分析,根据返回的结果知道函数的返回类型是vector<vector<int>>,传入的参数为candidates和target
- 由于是组合问题,为了使算法速度更快,因此将数据进行排序,
- 递归函数如何写,首先考虑的是传入的参数,candidates和target是肯定的,这里我需要一个sum来记录一条path下是数据总和,当然也可以不用,用target做减法等于0也是可以的,为了算法思路清晰我选择传入sum,需要一个index来作为遍历需要,作为for循环的起始位置。
- 对于组合问题,什么时候需要startIndex呢?
- 如果是一个集合来求组合的话,就需要startIndex
- 如果是多个集合取组合,各个集合之间相互不影响,那么就不用startIndex
- 递归结束条件:sum==target 递归结束,此时需要做的是将记录的path压入的result中。
- 接下来进行for循环,下标从index开始,进行剪枝操作,如果满足sum+candidates[i]<=target时,才进行操作,sum加上相应的元素,将当前元素压入到path中,然后进行递归操作,因为题目中可以重复选择元素,因此递归时index还是从i开始,然后进行回溯恢复原来的状态。
代码实现:
class Solution {
private:
vector<vector<int>> result;//全局变量记录结果
vector<int> path;//全局变量记录路径值
public:
vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
sort(candidates.begin(),candidates.end());
track(candidates,target,0,0);
return result;
}
void track(vector<int>& candidates, int target,int sum,int index){
if(sum==target){
result.push_back(path);
return;
}
for(int i=index;i<candidates.size();i++){
if(sum+candidates[i]<=target){//剪枝操作
sum+=candidates[i];
path.push_back(candidates[i]);
track(candidates,target,sum,i);
sum-=candidates[i];
path.pop_back();
}
}
}
};
没看明白,可以看:
「代码随想录」带你学透回溯算法!【39. 组合总和】 - 组合总和 - 力扣(LeetCode) (leetcode-cn.com)