找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。
说明:
所有数字都是正整数。
解集不能包含重复的组合。
示例 1:
输入: k = 3, n = 7
输出: [[1,2,4]]
示例 2:
输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum-iii
方法:这道题相比于leetcode40更简单一些。
我们只需要修改一处,就是当数字k已经用完的时候,但是总和差还不为0。
思路很简单,依旧l为索引位,我们需要判断两个参数,总和差和个数差。
class Solution {
public:
vector<vector<int>> combinationSum3(int k, int n) {
if( k * 9 < n) return res;
vector<int> t;
findcombinationSum(k,n,1,t);
return res;
}
private:
vector<vector<int>> res;
//l为索引位
void findcombinationSum(int k, int n, int l, vector<int> &t){
if(n == 0 && k == 0){
res.push_back(t);
return;
}
for(int i = l; i < 10; i++){
//个数已经用完,但是总和差不为0,直接break。
if(k == 0) break;
t.push_back(i);
//注意这里的索引是i+1
findcombinationSum(k-1,n-i, i+1,t);
t.pop_back();
}
return;
}
};