组合数问题
思路:
回溯算法的经典问题,回溯解决的是n层for循环嵌套的问题,如果说for循环是横向遍历的话,那么回溯就是纵向遍历,把组合数抽象为树的深度问题,n为树的宽度,k为树的深度,每次搜索树上的叶子结点,我们就找到了结果。
回溯的通用模板:
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
代码:
class Solution {
public:
vector<vector<int>> result;//所有路径
vector<int> path;//当前路径
vector<vector<int>> combine(int n, int k) {
back(n, k, 1);
return result;
}
void back(int n, int k,int index)
{
if (path.size() == k)//结束条件
{
return result.push_back(path);
}
for (int i=index; i <= n; i++)
{
path.push_back(i);//处理结点
back(n, k, i+1);//递归
path.pop_back();//回溯的关键,退回元素,撤销上一次的操作
}
}
};