回溯法思路总结【以Leetcode 78.子集、90. 子集II、22. 括号生成为例】【C++】

基本思想：

按一定路径遍历问题的所有可能结果，当路径符合要求时将得到的路径加入结果集后，回溯至上一步继续向后遍历。常用递归的方法实现代码。

特点：

当遇到需要得到所有可能结果的问题时可考虑使用。

存在的问题：

使用递归，若代码书写不规范或不够清晰时，可能需要耗费大量时间进行代码调试。

解决方法：

画递归树，思考人是怎么做的，找出共性规律。按照下述代码规范写法依次思考填充所需的部分。

规范写法（C++）：

class Solution {
public:
    void backtrack(vector<int>& nums, vector<vector<int>>& res, vector<int>& path, int start){
        //递归结束条件：将符合条件的路径加入结果集

        for(int i = start; i < nums.size(); i ++){//递归树每层需要走的选择
            //需要剪枝时增加剪枝条件
            //新一步的选择
            //递归，进入下一步
            //回溯
        }
    }
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<int> path; //路径
        vector<vector<int>> res; //结果集
        backtrack(nums, res, path, 0);
        return res;
    }
};

代入各例题思考，此处仅给代码及部分思考，具体思路使用上述规范写法进行思考填充即可完成：

78. 子集 - 力扣（LeetCode） (leetcode-cn.com)

class Solution {
public:
    void backtrack(vector<int>& nums, vector<vector<int>>& res, vector<int>& path, int start){
        res.push_back(path); //递归结束条件
        for(int i = start; i < nums.size(); i ++){
            path.push_back(nums[i]); //做出选择
            backtrack(nums,res, path, i+1); //递归
            path.pop_back(); //回溯
        }
    }
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<int> path;
        vector<vector<int>> res;
        backtrack(nums, res, path, 0);
        return res;
    }
};

90. 子集 II - 力扣（LeetCode） (leetcode-cn.com)

Leetcode 90相较78.子集多了需要剪枝的条件，我最开始写时未用规范写法写，发现剪枝处极容易出现遗漏情况，而按规范写法编写代码则可极大提高思考和调试效率。

class Solution {
    vector<vector<int>> res;
public:
    void backtrack(vector<int>& nums, vector<int>& path, int start){
        res.push_back(path);
        for(int i = start; i < nums.size(); i ++){
            if(i > start && nums[i - 1] == nums[i]) continue; //剪枝
            path.push_back(nums[i]);
            backtrack(nums,path,i+1);
            path.pop_back();
        }
    }
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        vector<int> path;
        sort(nums.begin(),nums.end());
        backtrack(nums, path, 0);
        return res;
    }
};

22. 括号生成 - 力扣（LeetCode） (leetcode-cn.com)

class Solution {
    vector<string> res;
public:
    void backtrack(string& path, int lnum, int rnum, int n){
        if(lnum == rnum && lnum == n){ //递归结束条件
            res.push_back(path);
            return;
        }
        //两种情况的选择、递归与回溯
        if(lnum < n){
            path.push_back('(');
            backtrack(path, lnum + 1, rnum, n);
            path.pop_back();
        }
        if(rnum < lnum){
            path.push_back(')');
            backtrack(path, lnum, rnum + 1, n);
            path.pop_back();
        }
    }
    vector<string> generateParenthesis(int n) {
        //画递归树，找条件；找结束条件push；回溯pop
        //对每一步，可以(，也可以)，)的数量应该小于等于(
        //如果没有(，则(,如果有剩余的)，可以走)，记录一下
        string p;
        backtrack(p, 0, 0, n);
        return res;
    }
};