找出所有相加之和为 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,为避免重复组合,规定数字只能往后取。例如,当前取了数字2,那么下一个数只能取2之后的数字。
代码展示:
class Solution {
public:
vector<vector<int>> combinationSum3(int k, int n) {
vector<int> path;
vector<vector<int>> ans;
int depth = 0;
int sum = 0;
int index = 1;
dfs(k,n,index,sum,depth,path,ans);
return ans;
}
void dfs(int k,int n,int index,int sum,int depth,vector<int>& path,vector<vector<int>>& ans){
if(depth==k && sum==n){
ans.push_back(path);
return;
}
if(depth>=k)
return;
for(int i=index;i<10;i++){
if(sum+i<=n){
path.push_back(i);
dfs(k,n,i+1,sum+i,depth+1,path,ans);
path.pop_back(); //回溯
}
}
}
};