题目(leecode T216):
找出所有相加之和为 n
的 k
个数的组合,且满足下列条件:
- 只使用数字1到9
- 每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
方法:本题和上一题组合的区别在于,本题的范围被指定为了1-9,即树的宽度被确定,总和n也被确定,深度k也是确定的。相当于在上一题的条件中加了几个限制,用的也是同样的回溯法。分析回溯的三要素得:
1:传入参数与返回值:回溯法的传入参数不是一次性就能确定的,需要我们在计算的过程中逐步完善。
- targetSum(int)目标和,也就是题目中的n。
- k(int)就是题目中要求k个数的集合。
- sum(int)为已经收集的元素的总和,也就是path里元素的总和。
- startIndex(int)为下一层for循环搜索的起始位置。
2:终止条件:因为我们所需要的元素个数为k,因此当path中存放了k个元素时,我们就可以结束了,然后判断当前path中的元素总和是否为n,如果是的话就作为一个结果存放下来,否则就直接返回即可。
3:单层回溯逻辑:在每层的处理过程中,我们要把新加入的元素i和当前的sum相加作为新的sum用来判断我们的sum的值是否达到了要求,同时注意回溯的过程,我们在加入一个元素并处理完之后还要把他给放出来,即减去当前值以便下一个值进来。
题解:
class Solution {
private:
vector<int> path;
vector<vector<int>> result;
void backtracking(int targetSum, int k, int sum, int startIndex){
if(path.size() == k){ //终止条件
if(sum == targetSum){
result.push_back(path);
}
return;
}
for(int i = startIndex; i<=9; i++){ //单层处理逻辑
sum += i;
path.push_back(i);
backtracking(targetSum,k,sum,i+1); //递归
sum -= i;
path.pop_back(); //回溯
}
}
public:
vector<vector<int>> combinationSum3(int k, int n) {
result.clear();
path.clear();
backtracking(n, k , 0, 1);
return result;
}
};