77、组合
第一次接触回溯的题目,回溯算法的模板框架如下:
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
本题也可以按照这个思路去做。
class Solution {
private:
vector<int> vec;
vector<vector<int>> result;
void backtracking(int n,int k,int startindex){
if(vec.size() == k){
result.push_back(vec);
return;
}
for(int i = startindex; i <= n;i++){
vec.push_back(i);
backtracking(n, k, i+1);
vec.pop_back();
}
}
public:
vector<vector<int>> combine(int n, int k) {
backtracking(n, k, 1);
return result;
}
};