题目描述
给定两个整数 n 和 k,返回 1 … n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
我的思路
- 继续死磕递归;
- 一看到这种求各种排列组合的问题,大概率使用递归比较方便。本题既然是要求输出的固定的数字个数,那么可以这么想,第一个数可以选择数组中任意一个数,第二个数选择第一个数之后的任意一个数。例如本题示例,第一个数可以选择 1-4 任意一个数,如果第一个数是 1,第二个数就是 2-4 任意一个数,如果第一个数是 2,第二个数就是 3-4 任意一个数,以此类推。(这样就能够解决输出的组合不会重复的问题,也不会有类似 1,1 | 2,2 这种组合);
- 因此 k 决定我们递归深度,n 决定我们的递归广度,当且仅当 k 为 0 时保存当前组合,每递归深一层 k - 1;
- 最重要的一点,每当递归完成一次,都应移除最后一个元素。否则就不会出现类似 1,2 | 1,3 | 1,4 | 2,3 …的组合,反而变成了 1,2 | 1,2,3 | 1,2,3,4 | 1,2,3,4,5 。
- 本题思路不难,也不容易出bug,所以通过率很高,但同样重要(在我看来)。
代码如下:
class Solution {
private:
vector<vector<int>> result;
vector<int> temp;
public:
vector<vector<int>> combine(int n, int k) {
if(k > n || n < 0 || k < 0) return result;
recur(1, n, k);
return result;
}
void recur(int index, int n, int k) {
if(k == 0){
result.push_back(temp);
return;
}
for(int i = index; i <= n; i++){
temp.push_back(i);
recur(i+1, n, k-1);
temp.pop_back();
}
return;
}
};