Given two integers n and k, return all possible combinations of k numbers out of 1 ... n.
For example,
If n = 4 and k = 2, a solution is:
[ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]
这道题其实就是列出C(n,k)的所有情况,难度为Medium。
比较直观的想法是采用递归回溯遍历每个数字取和不取的情况,递归结束条件是已经取到了k个数字,注意回溯之后需要修改的标记。具体代码:
class Solution {
public:
vector<vector<int> > combine(int n, int k) {
vector<vector<int>> rst;
vector<int> curRst;
if(k > 0 && n >= k)
getCombine(rst, curRst, 0, 0, n, k);
return rst;
}
void getCombine(vector<vector<int>> &rst, vector<int> &curRst, int pos, int cnt, int n, int k) {
if(cnt == k) rst.push_back(curRst);
else {
if(n-pos < k-cnt-1) return;
else {
while(pos < n) {
curRst.push_back(pos+1);
pos++;
cnt++;
getCombine(rst, curRst, pos, cnt, n, k);
cnt--;
curRst.pop_back();
}
}
}
}
};
另外还可以根据公式C(n,k) = C(n-1,k-1) + C(n-1,k),用纯递归的方法解决此问题。这里采用的其实是分治法,根据取与不取第n个数分为C(n-1,k-1)和C(n-1,k),然后将两部分合并。需要注意的是在得到C(n-1,k-1)之后要在每个条目中加入n,因为C(n-1,k-1)是默认选取n的,具体代码:
class Solution {
public:
vector<vector<int>> combine(int n, int k) {
vector<vector<int>> rst1, rst2;
if(n<k || k<0) return rst1;
if(n==k || k==0) {
vector<int> curRst;
for(int i=1; i<=k; i++) curRst.push_back(i);
rst1.push_back(curRst);
return rst1;
}
rst1 = combine(n-1, k-1);
for(int i=0; i<rst1.size(); i++)
rst1[i].push_back(n);
rst2 = combine(n-1, k);
copy(rst2.begin(), rst2.end(), back_inserter(rst1));
return rst1;
}
};