【代码随想录】回溯法

 前言

更详细的在大佬的代码随想录 (programmercarl.com)

本系列仅是简洁版笔记,为了之后方便观看

解决的类型问题

组合无序,排列有序

  1. 组合
  2. 切割
  3. 子集
  4. 排列(强调元素的顺序)
  5. 棋牌问题(n皇后,数独)

回溯法的三个步骤 

  1. 递归函数参数/返回值
  2.  终止条件
  3. 单层递归逻辑

去重词汇 

  1. 树层去重
  2. 树枝去重

模版

void backtracking(参数) {
    if (终止条件) {
        存放结果;
        return;
    }

    for (遍历结点) {
        处理节点;
        backtracking(路径,选择列表); // 递归
        回溯,撤销处理结果
    }
}

组合

 77. 组合 - 力扣(LeetCode)

class Solution {
    private:
    vector<vector<int>> result;
    vector<int>path;
    void backtracking(int n,int k,int a)//a表示递归的层数
    {
      if(path.size()==k)
      {
        result.push_back(path);
        return;
      }
      for(int i=a;i<=n;i++)
      {
            path.push_back(i); 
            backtracking(n, k, i + 1); 
            path.pop_back(); 
      }
    }
public:
    vector<vector<int>> combine(int n, int k) {
      backtracking(n,k,1);
      return result;
    }
};

 剪枝思路:从for循环开始的起始位置之后选择的元素个数不足需要的元素个数时,无需继续搜索

for (int i = a; i <= n - (k - path.size()) + 1; i++)

组合总和III

 和上一题整体模版是一样的,主要是多了sum的处理和sum的回溯 

216. 组合总和 III - 力扣(LeetCode)

class Solution {
private:
    vector<vector<int>>result;
    vector<int>path;
   void backtracking(int target,int k,int sum,int startIndex)
  {
    if(sum>target)
    {
        return;
    }
    if(path.size()==k)
    {
        if(sum==target)
        result.push_back(path);
        return ;
    }
    for(int i=startIndex;i<=9-(k-path.size())+1;i++)
    {
            sum += i; // 处理
            path.push_back(i); // 处理
            backtracking(target, k, sum, i + 1); // 注意i+1调整startIndex
            sum -= i; // 回溯
            path.pop_back(); // 回溯
    }
  }
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        result.clear(); // 可以不加
        path.clear();   // 可以不加
        backtracking(n, k, 0, 1);
        return result;
    }
};

电话号码的字母组合

17. 电话号码的字母组合 - 力扣(LeetCode)

1.用二维数组来做一个映射

2.对比发现隐藏的回溯

for (int i = 0; i < letters.size(); i++) {
            s.push_back(letters[i]);            
            backtracking(digits, index + 1);    
            s.pop_back();                      
        }
for (int i = 0; i < letters.size(); i++) {
            getCombinations(digits, index + 1, s + letters[i]);  
// void getCombinations(const string& digits, int index, const string& s)
        }

第二个是进入到下一层计算了,但是s没有变化,所以就相当于回溯了

class Solution {
private:
    const string letterMap[10]={
        //逐个映射
        "", // 0
        "", // 1
        "abc", // 2
        "def", // 3
        "ghi", // 4
        "jkl", // 5
        "mno", // 6
        "pqrs", // 7
        "tuv", // 8
        "wxyz"//9
    };
public:
    vector<string>result;
    string s;
    void backtracking(const string&digits,int index)
    {
       if (index == digits.size()) {
            result.push_back(s);
            return;
        }
        int digit=digits[index] - '0';//依次遍历数字
        string letters = letterMap[digit]; //寻找该位置对应的字符串     
        for (int i = 0; i < letters.size(); i++) {
            s.push_back(letters[i]);            
            backtracking(digits, index + 1);    
            s.pop_back();                       
        }

    }
    vector<string> letterCombinations(string digits) {
        s.clear();
        result.clear();
        if (digits.size() == 0) {
            return result;
        }
        backtracking(digits, 0);
        return result;
    }
};

组合总和

39. 组合总和 - 力扣(LeetCode)

1.和之前的组合总和最大的差别就是已经选取过的元素还可以再选取 

2.startIndex的作用是:每一次遍历开始的位置,使求出来的组合没有重复元素,例如组合23和组合32

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++) {
          sum += candidates[i];
          path.push_back(candidates[i]);
          backtracking(candidates, target, sum, i);//这儿不用i+1的原因是因为可以重复使用某个数字
          sum -= candidates[i];//回溯
          path.pop_back();

        }
    }
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        result.clear();
        path.clear();
        sort(candidates.begin(), candidates.end()); //从小到大排序,如果组合和大于所要求的值,则后面的数必定大于所要求的值
        backtracking(candidates, target, 0, 0);
        return result;
    }
};

 组合总和II

原数组有重复元素但结果不能有重复的组合(进行去重操作)

40. 组合总和 II - 力扣(LeetCode)

1.排序的原因:把相邻元素放在一起

2.本题去重关键:

树层去重,如果该层已经取过1了,那么它的同一层就不需要再取了,因为前面的1已经包含了1的所有组合(目的:没有重复的元素组合)

原数组有重复元素

树枝去重是没有必要的,因为一条链上本来就可以取得组合中的所有元素,也包括重复的元素(比如果该集合中有两个1,就可以同时取两个1)

3.相比上一题就是多了个used数组来记录该元素是否被使用过

4.used[i - 1] == false 表示,在同一树层上取的 candidates[i] 是从 candidates[i - 1] 回溯而来的,如果 used[i - 1] == true,就说明是进入下一层递归,去下一个数,就进行树枝上的深入

used[i - 1] == true,说明同一树枝candidates[i - 1]使用过(说明同一条链被用过)
used[i - 1] == false,说明同一树层candidates[i - 1]使用过

5.重点回溯代码

if (i > 0 && candidates[i] == candidates[i - 1] && used[i - 1] == false) {
                continue;
            }

 子集类问题可以不写终止条件

分割回文串

题目

131. 分割回文串 - 力扣(LeetCode)

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

思考结果

切割线:startIndex

截取的子串:[startIndex,i]

代码

class Solution {
private:
      vector<vector<string>> result;//二维数组
      vector<string>path;//定义一维数组
    bool istrue(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;
    }
    void backtracking(const string&s,int startIndex){
        if(startIndex>=s.size()){//结束条件
            result.push_back(path);
            return ;
        }
        for(int i=startIndex;i<s.size();i++){//从分割线开始
            if(istrue(s,startIndex, i)){//是回文子串
                string str=s.substr(startIndex,i-startIndex+1);//分割串的范围是[startIndex,i]
                 path.push_back(str);
            }
            else {                                // 不是回文,跳过
                continue;
            }
            backtracking(s, i + 1); //i+1为起始位置的回文子串
            path.pop_back(); //回溯
        }
    }
public:
    vector<vector<string>> partition(string s) {
        result.clear();
        path.clear();
        backtracking(s, 0);
        return result;
    }
};

复原IP地址

93. 复原 IP 地址 - 力扣(LeetCode)

pointNum记录逗号数量,pointNum=3说明此时有四个串

i+2的原因是在字符串中加入了分隔符.

class Solution {
private:
  vector<string>result;
  void backtracking( string& s,int startIndex,int pointnum){
    if(pointnum==3){
        if(isvalid(s,startIndex,s.size()-1)){
            result.push_back(s);
            return ;
        }
    }
         for (int i = startIndex; i < s.size(); i++) {
        if (isvalid(s, startIndex, i)) { 
                s.insert(s.begin() + i + 1 , '.');  
                pointnum++;
                backtracking(s, i + 2, pointnum);   
                pointnum--;                         
                s.erase(s.begin() + i + 1);  
        }else break;
    }
  }
      bool isvalid(const string& s, int start, int end) {
        if (start > end) {
            return false;
        }
        if (s[start] == '0' && start != end) { 
                return false;
        }
        int num = 0;
        for (int i = start; i <= end; i++) {
            if (s[i] > '9' || s[i] < '0') { 
                return false;
            }
            num = num * 10 + (s[i] - '0');
            if (num > 255) { 
                return false;
            }
        }
        return true;
    }
public:
    vector<string> restoreIpAddresses(string s) {
          result.clear();
          if(s.size()<4||s.size()>12) return result;
          backtracking(s,0,0);
          return  result;
    }
};

子集

78. 子集 - 力扣(LeetCode)

不同点:和之前学过的例子相比较,每一层递归里面都有自己需要的结果

终点状态:剩余集合是空集合

终止条件可以不加:因为如果大于了数组的大小,for循环不会进入进去,直接return

注意:把收获结果放在终止条件之前,不然会省略掉最后的结果

class Solution {
private:
    vector<vector<int>> result;
    vector<int> path;
    void backtracking(vector<int>& nums, int startIndex) {
       result.push_back(path);
       if (startIndex >= nums.size()) 
       { 
            return;
       }
        for (int i = startIndex; i < nums.size(); i++) {
            path.push_back(nums[i]);
            backtracking(nums, i + 1);
            path.pop_back();
        } 
    }   
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        result.clear();
        path.clear();
        backtracking(nums, 0);
        return result;
    }
};

子集II

90. 子集 II - 力扣(LeetCode)

原集合有重复元素但不能有重复子集

去重步骤:先排序,树层去重,树枝上不去重

 if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) {
                continue;
            }        

bool数组定义

 vector<bool> used(nums.size(), false);

总代码 

class Solution {
private:
    vector<vector<int>> result;
    vector<int> path;
    void backtracking(vector<int>& nums, int startIndex, vector<bool>& used) {
        result.push_back(path);
        for (int i = startIndex; i < nums.size(); i++) {
            if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) {
                continue;
            }
            path.push_back(nums[i]);
            used[i] = true;
            backtracking(nums, i + 1, used);
            used[i] = false;
            path.pop_back();
        }
    }

public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        result.clear();
        path.clear();
        vector<bool> used(nums.size(), false);
        sort(nums.begin(), nums.end()); 
        backtracking(nums, 0, used);
        return result;
    }
};

 子集、组合、分割问题的的区别

子集:收集树形结构中树的所有节点的结果。

组合、分割问题:收集树形结构中叶子节点的结果。

递增子序列

491. 非递减子序列 - 力扣(LeetCode)

给定一个整型数组, 你的任务是找到所有该数组的递增子序列,递增子序列的长度至少是2。

排序做法不可取:改变了元素的顺序

去重+非递减

  unordered_set<int> uset;
if ((!path.empty() && nums[i] < path.back())|| uset.find(nums[i]) != uset.end())
 {
                    continue;
 }

总代码

class Solution {
private:
    vector<vector<int>> result;
    vector<int> path;
    void backtracking(vector<int>& nums, int startIndex) {
        if (path.size() > 1) {//题意说大小要大于2
            result.push_back(path);
        }//不加return,因为它也属于子集问题
        unordered_set<int> used; // 每一次递归都要记录 仅是对于本层进行去重
        for (int i = startIndex; i < nums.size(); i++) {
         if ((!path.empty() && nums[i] < path.back())|| used.find(nums[i]) !=used.end()) 
            continue;
            used.insert(nums[i]); //本层已经用过该元素了
            path.push_back(nums[i]);//添加元素
            backtracking(nums, i + 1);//递归到下一层
            path.pop_back();//返回
        }
    }
public:
    vector<vector<int>> findSubsequences(vector<int>& nums) {
        backtracking(nums, 0);
        return result;
    }
};

全排列

used是标记该元素是否被用过

for循环从0开始不从之前的startIndex开始是因为排列讲究顺序,[1,2]和[2,1]是两种情况

class Solution {
public:
    vector<vector<int>>result;
    vector<int>path;
    void backtracking(vector<int>& nums, vector<bool>& used){
        if(path.size()==nums.size()){
            result.push_back(path);
            return;
        }
        for(int i=0;i<nums.size();i++){
            if(used[i]==true) continue;
            used[i]=true;
            path.push_back(nums[i]);
            backtracking(nums, used);
            path.pop_back();
            used[i] = false;
        }
    }
    vector<vector<int>> permute(vector<int>& nums) {
        result.clear();
        path.clear();
        vector<bool> used(nums.size(), false);
        backtracking(nums, used);
        return result;
    }
};

全排列II

47. 全排列 II - 力扣(LeetCode)

和上面的代码区别不大

 区别:

1.在for循环中加入下面的语句

if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false)
 {
                continue;
 }

2.在主函数中要先排序

sort(nums.begin(), nums.end()); 

N皇后

51. N 皇后 - 力扣(LeetCode)

class Solution {
private:
    vector<vector<string>>result;
    void backtracking(int n,int row,vector<string>& chessboard){
        if(row==n){
            result.push_back(chessboard);
            return ;
        }
        for(int col=0;col<n;col++){
            if(isvalid(row,col,chessboard,n)){
                chessboard[row][col]='Q';
                backtracking(n,row+1,chessboard);
                chessboard[row][col]='.';
            }
        }
    }
    bool isvalid(int row,int col,vector<string>& chessboard,int n){
        for (int i = 0; i < row; i++) { // 这是一个剪枝
        if (chessboard[i][col] == 'Q') {
            return false;
        }
    }
    // 检查 45度角是否有皇后
    for (int i = row - 1, j = col - 1; i >=0 && j >= 0; i--, j--) {
        if (chessboard[i][j] == 'Q') {
            return false;
        }
    }
    // 检查 135度角是否有皇后
    for(int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {
        if (chessboard[i][j] == 'Q') {
            return false;
        }
    }
    return true;
}


public:
    vector<vector<string>> solveNQueens(int n) {
        result.clear();
        vector<string> chessboard(n, string(n, '.'));
        backtracking(n, 0, chessboard);
        return result;
    }
};

解数独

37. 解数独 - 力扣(LeetCode)

 两个for循环

本题backtracking是bool类型,是因为只需要一个结果

class Solution {
private:
bool backtracking(vector<vector<char>>&board){
    for(int i=0;i<board.size();i++){
        for(int j=0;j<board[0].size();j++){
            if(board[i][j]=='.'){
                for(char k='1';k<='9';k++){
                    if(isvalid(i,j,k,board)){
                        board[i][j]=k;
                        if(backtracking(board)) return true;
                        board[i][j]='.';
                    }
                }
                return false;
            }
        }
    }
    return true;
} 
 bool isvalid(int row,int col,char val,vector<vector<char>>&board)  
 {
    for(int i=0;i<9;i++){
        if(board[row][i]==val){
            return false;
        }
    }
    for(int j=0;j<9;j++){
        if(board[j][col]==val){
            return false;
        }
    }    
    int startRow=(row/3)*3;
    int startCol=(col/3)*3;
    for(int i=startRow;i<startRow+3;i++){
        for(int j=startCol;j<startCol+3;j++){
          if (board[i][j] == val ) {
                return false;   
        }
      }
    }
    return true;

} 
public:
    void solveSudoku(vector<vector<char>>& board) {
         backtracking(board);
    }
};

  • 25
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值