给定两个整数 n 和 k,返回 1 … n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
第一次这么快的解出一个中等题,不过这个本身也很简单,但是要想到如何剪枝才能提高效率:
比如n=4,k=3的情况,{1,2,3,4}
要是递归到pos=3的情况,那根本不可能最后总数可以到3,这种情况应当被剪去
class Solution {
public:
vector<vector<int>> combine(int n, int k) {
vector<vector<int>> res;
vector<int> arr;
recruise(arr,res,1,n,k);
return res;
}
void recruise(vector<int> &arr,vector<vector<int>> &res,int pos,int n,int k)
{
if(arr.size()==k)
{
res.push_back(arr);
return;
}
if(n-pos+1<k-arr.size())
return;
//若剩下的可插入数都无法满足达到最后要求的数,那直接退出
for(int i=pos;i<=n;i++)
{
arr.push_back(i);
recruise(arr,res,++pos,n,k);
arr.pop_back();
}
}
};