【代码随想录算法训练Day24】LeetCode 77.组合

Day24 回溯

回溯算法的模板：

• 回溯函数模板返回值以及参数
• 回溯函数终止条件
• 回溯搜索的遍历过程

LeetCode 77.组合

回溯的经典问题，在每个路径数组中存入下一个数，再接着递归处理路径中的下一个数。

class Solution {
public:
    vector<vector<int>> res;
    vector<int> path;

    void backtrack(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);
            backtrack(n,k,i+1);
            path.pop_back();
        }
    }

    vector<vector<int>> combine(int n, int k) {
        backtrack(n,k,1);
        return res;
    }
};

剪枝

回溯本身是暴力的过程，可以看做是递归的循环，所以很多时候回溯是需要剪枝的，把某些不需要的或者不可能的回溯过程删除，达到提高回溯算法效率的目的。
本题的剪枝其实就是缩小遍历的范围，如果剩余元素已经不够我们所需要的集合元素个数，就可以立刻停止搜索，只需要把i的遍历范围修改了即可。

class Solution {
public:
    vector<vector<int>> res;
    vector<int> path;

    void backtrack(int n,int k,int startIndex){
        if(path.size()==k){
            res.push_back(path);
            return;
        }

        for(int i=startIndex;i<=n-(k-path.size())+1;i++){
            path.push_back(i);
            backtrack(n,k,i+1);
            path.pop_back();
        }
    }

    vector<vector<int>> combine(int n, int k) {
        backtrack(n,k,1);
        return res;
    }
};

经典的算法开始啦。