### 题目
给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
### 思路
就是一个回溯法,然后返回条件改成i<=n-(k-temp.size()),这样可以减少很多计算,因为temp必要要满足长度为k,如果i的长度过大这样i后面的元素并不足以构成一个长度为k的数组,就不用遍历了。
### code
class Solution {
public:
void huishu(int k,int n,int begin,vector<vector<int>>&res,vector<int>temp)
{
if(temp.size()==k)
{
res.push_back(temp);
return ;
}
for(int i=begin;i<=n-(k-temp.size());++i)
{
temp.push_back(i+1);
huishu(k,n,i+1,res,temp);
temp.pop_back();
}
return;
}
vector<vector<int>> combine(int n, int k) {
vector<vector<int>> res;
vector<int> temp;
huishu(k,n,0,res,temp);
return res;
}
};
分享一个别人的解法,这个写法挺有意思的。
首先是找到第一位可以的最小的数字,然后找第二位可以的最小数字,一直到第k位可以的最小数字,然后第k位开始不断增加,直到第k位已经到达最大,然后将第k-1位更改成第二小的数字,然后再依次增加第k位,如果第k-1位也不能增加了,在增加第k-2位,一直到第一位也不能再增加,那么循环结束。
class Solution {
public:
vector<vector<int>> combine(int n, int k) {
vector<vector<int>> result;
int i = 0;
vector<int> p(k, 0);
while (i >= 0) {
p[i]++;
if (p[i] > n) --i;
else if (i == k - 1) result.push_back(p);
else {
++i;
p[i] = p[i - 1];
}
}
return result;
}
};