回溯算法模板框架如下:
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
77.组合
class Solution {
public:
vector<int> path;
vector<vector<int>> res;
void backTracking(int n, int step, int k) {
if (k == 0) {
res.push_back(path);
return;
}
for (int i = step + 1; i <= n; i++) {
path.push_back(i);
backTracking(n, i, k - 1);
path.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
backTracking(n, 0, k);
return res;
}
};
剪枝
class Solution {
public:
vector<int> path;
vector<vector<int>> res;
void backTracking(int n, int step, int k) {
if (path.size() == k) {
res.push_back(path);
return;
}
for (int i = step; i <= n - (k - path.size()) + 1; i++) {
path.push_back(i);
backTracking(n, i + 1, k);
path.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
backTracking(n, 1, k);
return res;
}
};