Find all possible combinations of k numbers that add up to a number n, given that only numbers from 1 to 9 can be used and each combination should be a unique set of numbers.
Example 1:
Input: k = 3, n = 7
Output:
[[1,2,4]]
Example 2:
Input: k = 3, n = 9
Output:
[[1,2,6], [1,3,5], [2,3,4]]
思路:
这个问题还是比较简单的。在求解时我们把关注点放在给和等于n的k项组合选择第一项上,这样,加入我们已经选择好了第一项x,剩下的问题就是给和等于n-x的k-1项组合选择第一项。另外,由于本题还有其他约束:1.不能重复;2.组合内元素升序排列;因此可以更具体的说剩下的问题是在大于x的数中给和等于n-x的k-1组合选择第一项,这样直到k-1为0且n-x为0时,就找到了一个可行的组合。
我们可以编写一个函数,每次都是选择组合内的某一项,并递归调用来选择下一项。需要特别说明的一点是,为了保证数字不重复,我们通常会使用一个标志数组来记录数字的使用情况,然而我们这里不用这么做。我们可以通过在大于x的数中选择这个约束来保证,递归选择下一项时是在从未使用过的数字中选择的,一定不会重复,而且这样还能保证每一项添加到组合中时是以升序排列的。
代码1:
class Solution {
public:
vector<vector<int>> findaim(int head, int k, int n) {
if (k == 1) {
if (head<n && n<10) {
vector<vector<int>> result;
result.resize(1);//r行
result[0].resize(1);//每行为c列
result[0][0] = n;
return result;
}
else {
vector<vector<int>> result;
return result;
}
}
vector<vector<int>> jj;
for (int i = head+1; i<10; i++) {
vector<vector<int>> j = findaim(i, (k - 1), (n - i));
int row = j.size();
if (row != 0) {
for (int ii = 0; ii<row; ii++) {
j[ii].push_back(i);
}
jj.insert(jj.end(), j.begin(), j.end());
}
}
return jj;
}
vector<vector<int>> combinationSum3(int k, int n) {
if (k == 1 && n<10) {
vector<vector<int>> result;
result[0][0] = n;
return result;
}
vector<vector<int>> jj;
for (int i = 1; i<10; i++) {
vector<vector<int>> j = findaim(i, (k - 1), (n - i));
int row = j.size();
if (row != 0) {
for (int ii = 0; ii<row; ii++) {
j[ii].push_back(i);
}
jj.insert(jj.end(), j.begin(), j.end());
}
}
return jj;
}
};
代码2:
class Solution {
public:
vector<vector<int>> helper(int k, int n, int begin){
vector<vector<int>> res;
if(k<1 || n<1 ||begin>9){return res;}
for(int i=begin;i<10;i++){
vector<vector<int>> temp;
if((n-i)==0 && k==1){
res={{i}};
return res;
}
else{
temp = helper(k-1, n-i, i+1);
if(temp.size()!=0){
for(vector<int> sub:temp){
sub.insert(sub.begin(),i);
res.push_back(sub);
}
}
}
}
return res;
}
vector<vector<int>> combinationSum3(int k, int n) {
return helper(k,n,1);
}
};
代码3;
class Solution {
public:
vector<vector<int>> combinationSum3(int k, int n) {
vector<vector<int>> res;
vector<int> solu;
combination(res, solu, k,n);
return res;
}
void combination(vector<vector<int>> & res, vector<int> solu, int k, int n) {
if (solu.size() == k && n == 0) res.push_back(solu);
if (solu.size() < k) {
for (int i = solu.size() == 0 ? 1 : solu.back() + 1; i <=9; i++) {
if (i > n) break;
solu.push_back(i);
combination(res, solu, k, n -i);
solu.pop_back();
}
}
}
};