常见的回溯算法面试题
回溯算法解题模板
对于用回溯解决的题目,代码大部分可以用下面的模板来套:
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
<return_type> solution(参数) {
backtracking(路径,选择列表);
return result;
}
以77.组合这个题为例:
class Solution {
public:
void back_trace(int n, int k, int startIndex) {
if (path.size() == k) {
// 递归出口,当满足条件时退出递归
result.push_back(path);
return;
}
for (int i = startIndex; i <= n; ++i) {
path.push_back(i);
back_trace(n, k, i + 1); // 递归的放入后面的数字
path.pop_back(); // i这次循环的递归结束,把这词的结果删掉,等下次循环放入下次的结果
}
}
vector<vector<int>> combine(int n, int k) {
// result.clear();
// path.clear();
back_trace(n, k, 1);
return result;
}
private:
vector<vector<int>> result; // 用于存放结果
vector<int> path; // 结果的一个item
};