算法|回溯 03

LeetCode 39- 组合总和

题目链接：力扣

题目描述：给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ，找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ，并以列表形式返回。你可以按 任意顺序 返回这些组合。

candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同，则两种组合是不同的。 

对于给定的输入，保证和为 target 的不同组合数少于 150 个。

解题思路

• 经典的回溯题目，本题要求我们可以重复取数字，但是不能有重复的组合。

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

    void backtracking(vector<int> &nums,int target,int sum,int startIndex){

        //和大于目标值不符合条件直接返回
        if(sum > target){
            return ;
        }

        //等于目标值，收集结果
        if(target == sum){
            result.push_back(path);
            return;
        }

        //循环递归（这个循环其实就是把所有数字拿来组合）
        for(int i=startIndex;i<nums.size();i++){
                //递归
                sum += nums[i];
                path.push_back(nums[i]);

                //由于每个数字可以重复取，所以我们可以使得startIndex = i,下一次进入递归的时候
                //还是从candidates[i]开始选，这样就使得重复使用数字
                backtracking(nums,target,sum,i);

                //回溯
                sum -= nums[i];
                path.pop_back();
        }
    }

    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        backtracking(candidates,target,0,0);
        return result;
    }
};

总结：

• for循环可以剪枝（先排序），然后再这样剪枝即可

for (int i = startIndex; i < candidates.size() && sum + candidates[i] <= target; i++)

所以有了如上的剪枝操作，我们每次进行剪枝一般都是在for循环中下功夫

LeetCode 40- 组合总和II

题目链接：力扣

题目描述：给定一个候选人编号的集合 candidates 和一个目标数 target ，找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用 一次 。

注意：解集不能包含重复的组合。 

解题思路

• 这题和上一题的区别就是元素不能重复使用，但是如果有两个元素相同，那他们是可以出现在同一个集合里面的。
• 这题关键就难在怎么去除重复的组合。首先我们进行排序，那样相同的元素就在相邻的位置了。由于我们每次是从startIndex开始取的，所以两个元素相同时，我们是这样取的：

看看used数组怎么发挥作用（图有点难画，直接借用卡哥的）

class Solution {
private:
    vector<vector<int>> result;
    vector<int> path;
    void backtracking(vector<int>& candidates, int target, int sum, int startIndex, vector<bool>& used) {
        if (sum == target) {
            result.push_back(path);
            return;
        }
        for (int i = startIndex; i < candidates.size() && sum + candidates[i] <= target; i++) {
            // used[i - 1] == true，说明同一树枝candidates[i - 1]使用过
            // used[i - 1] == false，说明同一树层candidates[i - 1]使用过
            // 要对同一树层使用过的元素进行跳过
            if (i > 0 && candidates[i] == candidates[i - 1] && used[i - 1] == false) {
                continue;
            }
            sum += candidates[i];
            path.push_back(candidates[i]);
            used[i] = true;
            
            backtracking(candidates, target, sum, i + 1, used); // 和39.组合总和的区别1，这里是i+1，每个数字在每个组合中只能使用一次
            
            used[i] = false;
            sum -= candidates[i];
            path.pop_back();
        }
    }

public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        vector<bool> used(candidates.size(), false);
        path.clear();
        result.clear();
        // 首先把给candidates排序，让其相同的元素都挨在一起。
        sort(candidates.begin(), candidates.end());
        backtracking(candidates, target, 0, 0, used);
        return result;
    }
};

总结：

• 去重的逻辑，有点难想到，这是这题的难点。

• if(i>start && candidates[i]==candidates[i-1]) continue;这个去重方法可以直接不用used数组，这个方法的原理就是出现重复的树枝时只保留最左侧的树枝，别的全部都continue跳过。比如说 【1， 1， 1， 1， 2】 求和是3， 只保留最左侧的 1->1->1这样子一支树。当前前面的1进入，肯定都是依靠递归，那样start 都是 和i相等的。（其实这个条件没太看懂）

LeetCode 131- 分割回文串

题目链接：力扣

题目描述：给你一个字符串 s，请你将 s 分割成一些子串，使每个子串都是 回文串 。返回 s 所有可能的分割方案。

回文串 是正着读和反着读都一样的字符串。

解题思路

• 首先我们用startIndex来控制当前切割到哪个字符了，当切割到最后一个字符时，说明到头了，返回结果即可。
• 我们要单独写一个判断函数来判断是否为回文子串。
• 进入递归时首先判断当前要切割的字串是不是回文子串，是就添加，不是就continue，下一次递归再进入到i+1的位置进行切割，我们每次切割的当前字串是区间是[startIndex,i]所以每次需要这样判断即可。

class Solution {
private:
    vector<vector<string>> result;
    vector<string> path; // 放已经回文的子串
    void backtracking (const string& s, int startIndex) {
        // 如果起始位置已经大于s的大小，说明已经找到了一组分割方案了
        if (startIndex >= s.size()) {
            result.push_back(path);
            return;
        }
        for (int i = startIndex; i < s.size(); i++) {
            if (isPalindrome(s, startIndex, i)) {   // 是回文子串
                // 获取[startIndex,i]在s中的子串
                string str = s.substr(startIndex, i - startIndex + 1);
                path.push_back(str);
            } else {                                // 不是回文，跳过
                continue;
            }
            backtracking(s, i + 1); // 寻找i+1为起始位置的子串
            path.pop_back(); // 回溯过程，弹出本次已经添加的子串
        }
    }
    bool isPalindrome(const string& s, int start, int end) {
        for (int i = start, j = end; i < j; i++, j--) {
            if (s[i] != s[j]) {
                return false;
            }
        }
        return true;
    }
public:
    vector<vector<string>> partition(string s) {
        result.clear();
        path.clear();
        backtracking(s, 0);
        return result;
    }
};

总结：这题用start到i来控制切割有点难想到。