题目描述:
给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
思路:回溯法,从第一个数开始遍历,组合中可以包括这个数,然后递归求得从2到n中k-1个数的组合;组合中也可以不包括这个数,列举完包括这个数的情况再递归求得从2到n中k个数的组合。所以首先需要一个变量标识求从几到n的组合,还需要k变化表示还要求几个数。k=0时表示一个组合列举完,存入结果中。
注意:k-1和i+1都要在调用函数里面,而不能先将k--,i++后再调用,因为当组合中不包括这个数时,还是要求k个数的组合。如果先将k--,temp.pop_back()之后k就变了。
AC代码
class Solution {
public:
vector<vector<int> > combine(int n, int k) {
vector<vector<int> > result;
if(n<1||k<1||k>n)
return result;
vector<int> temp;
combineCore(n, k, result, temp, 1);
return result;
}
void combineCore(int n, int k, vector<vector<int> >& result, vector<int>& temp, int start)
{
if(k==0)
{
result.push_back(temp);
return;
}
for(int i=start;i<=n-k+1;i++)
{
temp.push_back(i);
combineCore(n, k-1, result, temp, i+1);
temp.pop_back();
}
}
};
刚开始一直有错?
class Solution {
public:
vector<vector<int> > combine(int n, int k) {
vector<vector<int> > result;
if(n<1||k<1||k>n)
return result;
vector<int> temp;
combineCore(n, k, result, temp, 1);
return result;
}
void combineCore(int n, int k, vector<vector<int> >& result, vector<int>& temp, int start)
{
if(k==0)
{
result.push_back(temp);
return;
}
for(int i=start;i<=n-k+1;i++)
{
temp.push_back(i);
k--;
combineCore(n, k, result, temp, i+1);
temp.pop_back();
}
}
};