Leetcode 216. 组合总和 III

题目

找出所有相加之和为 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]]

题解

dfs回溯
用一个变量记录选中的数的和，如果大于目标值或大于k个数，则返回；等于，且正好选了k个数，则放入答案中；小于，则接着进行选择。详细过程见代码

代码

vector<vector<int>> ans;
void dfs(int begin,int time,int k,int sum,int n,vector<int>& now){
if(time == k && sum==n){	//选了k个数，和为n，则把选的数放入答案中
ans.push_back(now);
}else if(time < k){
if(sum >= n)    return;		//和大于目标值，返回；等于目标值，再选一个数肯定会大于目标值，也返回
for(int i=begin; i<10; i++){
now.push_back(i);
dfs(i+1,time+1,k,sum+i,n,now);
now.pop_back();
}
}
}
vector<vector<int>> combinationSum3(int k, int n) {
vector<int> now;
dfs(1,0,k,0,n,now);
return ans;
}

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/combination-sum-iii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。