代码随想录算法训练营day23

5 阅读2分钟

回溯算法的进一步强化 39.组合总和 和之前的组合数类似,仅仅需要注意可以取重就可以


class Solution {
private:
    vector<vector<int>> result;//存储所有结果
    vector<int> tmp;//存储临时结果
    void backtracking(vector<int>& candidates,int target,int index,int sum){
        //递归底层
        if(sum>target) return;
        else if(sum==target) {
          result.push_back(tmp);
          return;
        }
        //单层递归逻辑
        for(int i=index;i<candidates.size()&&sum<target;i++){
          sum=sum+candidates[i];
          tmp.push_back(candidates[i]);
          backtracking(candidates,target,i,sum);
          sum=sum-candidates[i];
          tmp.pop_back();
        }
    }
    
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        result.clear();
        tmp.clear();
        backtracking(candidates,target,0,0);
        return result;
        
    }
};

40 组合数二 增加了剪支操作以及判断不能够重复取数,减少时间复杂度

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

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

131 分割回文串

class Solution {
//设计思路,采用递归的方式不断对s进行拆分递归。
//第一步,递归函数需要传入的参数,除s之外,还应当有分割下标。具体结果分别存储在result和tmp之中
private:
    vector<vector<string>> result;
    vector<string> tmp;
    //判断字符串是否为回文串的函数
   bool reword(string s){
          //使用双指针
        for( int i=0,j=s.size()-1;i<j;i++,j--){
            if(s[i]!=s[j])
               return false;
        }
        return true;
    }
   
   void backtracking(string s,int index){
        //底层递归,当index==s.size()时表示回文串被分割完,返回
        if(index==s.size()){
            result.push_back(tmp);
            return;
        }
        //单层递归逻辑
        for(int i=index;i<s.size();i++){
            //判断该段分割是否满足回文字符串
            string cur=s.substr(index, i - index + 1);
            if(!reword(cur))
               continue;
            tmp.push_back(cur);
            backtracking(s,i+1);
            tmp.pop_back();
        }
    }
public:
    vector<vector<string>> partition(string s) {
        result.clear();
        tmp.clear();
        backtracking(s,0);
        return result;
    }
};