代码随想录算法训练营第24天|回溯算法、77组合
77.组合
这就是一个简单的组合问题。如果使用的是直接用for循环套暴力穷举法,一旦k的值过大,for循环也不行了。
回溯法的模板:
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
使用模板:
class Solution {
private:
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; ++i) {
path.push_back(i);
backtracking(n, k, i + 1);
// 回溯
path.pop_back();
}
}
public:
vector<vector<int>> combine(int n, int k) {
res.clear();
path.clear();
backtracking(n, k, 1);
return res;
}
};