回溯
理论基础
本质是暴搜,配有剪枝会快
代码模板
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
77. 组合
思路
- 截止条件:
path
中个数为k
就放到结果里 - 横向遍历:组合问题要去除自己,还要去除遍历过的,无序问题,所以再找应该从自己之后的找
代码
class Solution {
public:
vector<vector<int>> res;
vector<int> path;
vector<vector<int>> combine(int n, int k) {
backtracking(1, n, k);
return res;
}
void backtracking(int s, int n, int k){
if (path.size() == k){
res.push_back(path);
return;
}
// for (int i = s; i <= n; i ++){ 优化剪枝
for (int i = s; i <= n - (k - path.size()) + 1; i ++){
path.push_back(i);
backtracking(i + 1, n, k);
path.pop_back();
}
}
};