描述
给定两个整数 n 和 k,返回 1 … n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combinations
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
和全排列的思路是基本一致的,简化版
就是
backtrack(left,cur,n,k)
{
if(n==k)
{ans.push_back(cur);return}
for(int i = left;i<=n;i++)
{
cur.push_back(nums[i]);
backtrack(i+1,cur,n,k);
cur.pop_up();
}
}
但是这样速度是比较慢的,可以进行优化
我们是要在n个数里找有k个数的组合
如果当前n=5,cur.size() ==2,i = 4,k=4,还需要k-cur.size()-2个数,当当前循环只有1个数了,不可能有k=4的组合,所以这个循环是没有必要的
也就是循环条件不是i<=n,而是i<=n-(k-cur.size()),才能保证当前分支有足够的数可以填充到集合里
代码
class Solution {
public:
vector<int>nums;
vector<vector<int>>ans;
void backtrack(vector<int>cur, int n, int k, int left)
{
if (cur.size() == k)
{
ans.push_back(cur);
return;
}
//for (int i = left; i < n; i++)
for(int i = left; i<=n-(k-cur.size());i++)
{
int nu = nums[i];
cur.push_back(nu);
backtrack(cur, n, k, i + 1);
cur.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
for (int i = 1; i <= n; i++)
nums.push_back(i);
vector<int>cur;
backtrack(cur, n, k, 0);
return ans;
}
};