leetcode 77 组合 回溯 C++

描述

给定两个整数 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;
	}
};