给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
解题思路
回溯三部曲:
1.递归函数返回类型和传入参数
2.回溯终止条件
3.单层搜索的逻辑
class Solution {
public:
vector<vector<int>> res;
vector<int> path;
void backtracking(int n, int k, int startIndex){ //startIndex控制开始搜索的元素位置
if(path.size() == k){ //终止条件,k控制path里面数的个数
res.push_back(path);
return;
}
for(int i=startIndex; i<=n; i++){
path.push_back(i); //处理节点
backtracking(n, k, i+1); //递归
path.pop_back(); //回溯,撤销处理的节点
}
}
vector<vector<int>> combine(int n, int k) {
backtracking(n, k, 1);
return res;
}
};
作者:jasscical
链接:https://leetcode-cn.com/problems/combinations/solution/77zu-he-by-jasscical-2/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
参考-- 剪枝操作
```cpp []
//剪枝写法
class Solution {
public:
vector<vector<int>> res;
vector<int> path;
void backtracking(int n, int k, int startIndex){
if(path.size() == k){
res.push_back(path);
return;
}
for(int i=startIndex; i<=n-(k-path.size())+1; i++){ //剪枝操作
path.push_back(i);
backtracking(n, k, i+1);
path.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
backtracking(n, k, 1);
return res;
}
};
```