@代码随想录算法训练营第24天 | 回溯算法理论基础,LeetCode77. 组合
回溯算法理论基础
回溯算法本质还是暴力迭代,只不过有的时候for循环的暴力迭代是无法写出来的,所以需要用回溯算法,借助递归实现。
比如下面这道题:
给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
这种就需要k个loop,当k很大的时候几乎无法用for循环实现,那么就要用到回溯算法。
本质也是一个n叉树:
下面是一套回溯的代码模板:
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
77. 组合
第一遍读题思考
比较典型
代码随想录解法思路
套模版
c++代码具体实现注意事项
自己写的代码,可能比代码随想录里的不够精简,但是运行效果差不多。主要是为了方便理解。
(递归版本)
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
void trversal(int n, int k, int start){
if(path.size()==k){
result.push_back(path);
return;
}
for(int i=start;i<=n;i++){
path.push_back(i);
trversal(n,k,i+1);
path.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
result.clear();
path.clear();
trversal(n,k,1);
return result;
}
};