版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/u010653518/article/details/84197852
找出所有相加之和为 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]]
执行用时为0ms的
class Solution {
public:
vector<vector<int>> combinationSum3(int k, int n) {
vector<vector<int>> result;
if(n < 3 || n > 45) return result;
vector<int> temp;
for(int i = 1;i <= 9;i++) {
dfs(result,temp,i,0,1,k,n);
}
return result;
}
void dfs(vector<vector<int>> & result,vector<int> &temp,int cur,int sum,int index,int k,int n) {
temp.push_back(cur);
sum += cur;
if(sum == n && index == k) {
result.push_back(temp);
temp.pop_back();
sum -= cur;
return;
}
else if(index == k) {
temp.pop_back();
sum -= cur;
return;
}
for(int i = cur+1;i < 10;i++) {
dfs(result,temp,i,sum,index+1,k,n);
}
temp.pop_back();
sum -= cur;
return;
}
};
我的解答:
class Solution {
public:
vector<vector<int> > combinationSum3(int k, int n)
{
vector<vector<int> > res;
vector<int> path;
int a=1,d=1;
dfs(res,path,k,n,a,d);
return res;
}
void dfs(vector<vector<int> >& res,vector<int>& path,int k,int n,int a,int d)
{
if(n<a||a>9)
{
return;
}
if(d==k&&n<10)
{
path.push_back(n);
res.push_back(path);
path.pop_back();
return;
}
while(a<10)
{
path.push_back(a);
dfs(res,path,k,n-a,a+1,d+1);
path.pop_back();
a++;
}
}
};
#发现一个错误测试用例:
1
10