代码随想录 第七章 回溯算法part02 39. 组合总和 40.组合总和II 131.分割回文串

39. 组合总和

class Solution {
private:
    vector<vector<int>>result;
    vector<int>path;
    void bt(vector<int>& candidates, int target, int i){
        if(target<0) return;
        if(i==candidates.size()&&target!=0) return;
        if(target==0){
            result.push_back(path);
            return;
        }
        while(i<candidates.size()){
            path.push_back(candidates[i]);
            bt(candidates,target-candidates[i],i);
            path.pop_back();
            i++;
        }
    }
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        bt(candidates,target,0);
        return result;
    }
};

简单的。

40.组合总和II

class Solution {
private:
    vector<vector<int>>result;
    vector<int>path;
    void bt(vector<int>& candidates, int target, int i){
        if(target<0) return;
        if(i==candidates.size()&&target!=0) return;
        if(target==0){
            result.push_back(path);
            return;
        }
        while(i<candidates.size()){
            path.push_back(candidates[i]);
            bt(candidates,target-candidates[i],i+1);
            path.pop_back();
            while(i<candidates.size()-1){
                if(candidates[i]!=candidates[i+1]) break;
                i++;
            }
            i++;
        }
    }
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        sort(candidates.begin(),candidates.end());
        bt(candidates,target,0);
        return result;
    }
};

相较于上一题，这题的区别在于，所给的数组不仅时乱序，并且有相同的数字重复出现，要保证组合不重复，可以对数组先进行排序，在回溯后，加一个检测下一格数字是否与当前数字相同的循环，避免出现重复组合。

131.分割回文串

class Solution {
private:
    vector<vector<string>> result;
    vector<string> path;
    void bt(string s){
        if(s.size()==0){
            result.push_back(path);
            return;
        }
        for(int i=1;i<=s.size();i++){
            int j=0,k=i-1;
            while(j<=k){
                if(s[j]!=s[k]) break;
                j++;
                k--;
            }
            if(j>k){
                path.push_back(s.substr(0,i));
                bt(s.substr(i,s.size()));
                path.pop_back();
            }
        }
    }
public:
    vector<vector<string>> partition(string s) {
        bt(s);
        return result;
    }
};

这题可以通过substr对字符串逐层分割，每层bt对拿到的字符串从第一位开始寻找是否有回文片段，找到符合回文性质的片段后，将剩余片段传入下一层bt，直到分割出的字符串为空。

代码随想录 第七章 回溯算法part02