回溯算法:回溯入门,组合数问题

组合数问题
在这里插入图片描述
思路:
在这里插入图片描述

回溯算法的经典问题,回溯解决的是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();//回溯的关键,退回元素,撤销上一次的操作
        }
    }
};

猜你喜欢

转载自blog.csdn.net/cckluv/article/details/111589926