百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 博客教程 > 正文

面试必备:回溯算法详解

connygpt 2024-12-20 11:48 2 浏览

我们刷leetcode的时候,经常会遇到回溯算法类型题目。回溯算法是五大基本算法之一,一般大厂也喜欢问。今天跟大家一起来学习回溯算法的套路,文章如果有不正确的地方,欢迎大家指出哈,感谢感谢~

  1. 什么是回溯算法?
  2. 一道算法题走进回溯算法
  3. 回溯算法框架套路
  4. leetcode案例分析

1. 什么是回溯算法

回溯算法,一种通过探索所有可能的候选解来找出所有的解的算法

它采用试错的思想,它尝试分步地去解决一个问题。在分步解决问题的过程中,当它通过尝试发现现有的分步答案不能得到有效的正确地解答的时候,它将取消上一步甚至是上几步的计算,再通过其它的可能的分步解答再次尝试寻找问题的答案。回溯法通常用最简单的递归方法来实现,在反复重复上述的步骤后可能出现两种情况:

  • 找到一个可能存在的正确的答案;
  • 在尝试了所有可能的分步方法后宣告该问题没有答案。

举个类似的生活例子,比如放羊娃的羊在分岔路口走丢了,他顺着不同的岔路口寻找羊,一个岔路口一个岔路口的去尝试找羊。如果找不到羊,继续返回来找到岔路口的另一条路,直到找到羊为止。

如下图为找羊的决策路线图:

放羊娃在A方向找,然后走C方向,没找到时,他回到分岔路,又朝D方向走...直到找到羊,这就是回溯

2. 一道算法题走进回溯算法

给定一个不含重复数字的数组 nums ,返回其所有可能的全排列。你可以 按任意顺序 返回答案。

示例 1:

输入:nums = [1,2,3]
输出:[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]

实例2:

输入:nums = [0,1]
输出:[[0,1],[1,0]]

2.1 实现思路

看完这道题,我们的第一想法就是穷举全排列呀,但是你不会毫无规律得穷举对吧。比如要排3个数[1,2,3],你会第一位先排1,然后第二位只能是2或者3,如果第二位是2,第三位只能是3了...

我们可以借鉴羊娃找羊的路线图,画出全排列的树图,如下:

其实我们是不是从根节点遍历这棵树,记录下走过路径的数字,走到叶子节点,就可以得到一个排列啦。走完所有的叶子节点,那就可以得到全排列啦。

这棵树如何理解更清晰呢?

  • 为了方便理解,我们可以把nums个数k看做k种选择,比如对于[1,2,3],每一位都有3种选择:1、2、3。
  • 每一次做选择,都展开出一棵空间树,
  • 选择完后,如果是重复选的路径,就做剪枝。

上图的那棵树,可以看做是遍历所有元素,展开空间树,然后剪枝得来的。如下图

好啦,现在知道树怎么来的,我们来看下怎么遍历找到全排列呢?每次走树的分支,都像是在做决策。我们可以把已走的路径可做的选择作为树节点的两个属性。

如果在根节点,可做的选择为1、2、3,走过的路径为空,如下图

走到叶子节点时,已走路径数组长度等于原素组的个数,这时候走过路径就是满足条件的一个解。

2.2 代码实现

代码怎么写呢?以前我们学习树的遍历,一般都用到递归,这道题也用递归。

  • 递归的入口是什么呢?一个可选路径和一个已走过的路径就好啦。
  • 递归函数体呢?一个for循环,枚举当前数组的元素,并且需要if判断,以跳过剪枝
  • 递归出口呢?也就是走到叶子节点啦,叶子节点,就是当构建的已走路径path的数组长度等于nums的长度

实现代码如下:


class Solution {
    //全排列,即所有路径集合
    List<List<Integer>> allPath = new LinkedList<>();

    public List<List<Integer>> permute(int[] nums) {
        //当前路径,入口路径,path是空的
        List<Integer> path =  new LinkedList<>();
        //递归函数入口,可做选择是nums数组
        backTrace(nums,path);
        return allPath;
    }

    public void backTrace(int[] nums,List<Integer> path){
        //已走路径path的数组长度等于nums的长度,表示走到叶子节点,所以加到全排列集合
        if(nums.length==path.size()){
           allPath.add(new LinkedList(path));
           return;
        }

        for(int i=0;i<nums.length;i++){
            //剪枝,排查已经走过的路径
            if(path.contains(nums[i])){
                continue;
            }
            //做选择,加到当前路径
            path.add(nums[i]);
            //递归,进入下一层的决策
            backTrace(nums,path);
            //取消选择
            path.remove(path.size() - 1);
        }
    }
}

为什么要回溯呢?或者说为什么用到回溯算法呢?

  • 因为我们不是要找到一个排列就好了,而是需要找出所有满足条件的排列
  • 当递归调用结束时,结束的是当前的递归分支,还需要去别的分支继续找
  • 因此需要撤销当前的选择,回到选择前的状态,再选下一个选项,即进入下一个分支。

3. 回溯算法框架套路

  1. 穷举找规律,总结出回溯决策树
  2. 套用回溯算法框架代码

3.1. 穷举找规律,总结出回溯决策树

回溯类型问题的问题,基础也是穷举。我们一般通过穷举找到规律,然后画出回溯决策树就好啦。比如以上全排列的例子。

决策树的节点一般有两个属性,就是已走路径已经可做的选择。在总结决策回溯树的时候需要关注下。

3.2. 套用回溯算法框架

决策一个回溯问题,实际上就是解决一个决策树的遍历过程。需要考虑这三个问题:

  • 已走路径:已做出选择,走过的路径
  • 可选列表:你当前可以做的选择
  • 结束条件:一般走到决策树的叶子节点,它无法再做别的条件选择

回溯算法伪代码框架如下:

//所有路径集合
List<> allPath  = []
void backTrace (可选列表,已走路径):
     if(满足结束条件){
        allPath.add(已走路径);
        return;
     }
     for(选择:可选列表){
        做选择
        backTrace(当前可选列表,已走路径);
        撤销选择
     }

4. leetcode案例分析

题目:

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

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

示例1:

输入:candidates = [2,3,6,7], target = 7
输出:[[2,2,3],[7]]
解释:
2 和 3 可以形成一组候选,2 + 2 + 3 = 7 。注意 2 可以使用多次。
7 也是一个候选, 7 = 7 。
仅有这两种组合。

实例2:

输入: candidates = [2,3,5], target = 8
输出: [[2,2,2,2],[2,3,3],[3,5]]

4.1 思路

我们先穷举找下规律嘛,拿示例1的数据candidates = [2,3,6,7], target = 7

7 = 2 + 2 + 3
7 = 7

再拿示例2的数据:

8 = 2+ 2 + 2 +2
8 = 2 + 3 + 3
8 = 3 + 5

其实规律还是比较清晰的,我们只需要把target一个一个减去candidates数组的元素,如果可以减到为0,那么就是一个解。

接下来我们就是画树啦,可以把target当做树的根节点,然后分支分别表示减去candidates数组的中元素,然后子节点就是target减去数组元素的差,如下:

接下来我们可以套用回溯算法框架啦,已走路径,可选列表,结束条件分别怎么表示呢?看下下图:

如果走到橙色节点4这个位置,可选列表就是减2,或者减3啦,因为减6的话为负数啦。怎么确定是可选列表呢?只要当前的target减去要选分支的值大于0,都可以作为可选列表。

已走路径就是 -3这个分支

结束条件呢?当走到负数或者0的节点,都表示该结束啦,已经无法决策啦。

4.2 代码实现

最后我们套下溯算法框架伪代码,如下:

class Solution {

    List<List<Integer>> allPath = new LinkedList<>();

    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        List<Integer> path = new LinkedList<>();
        backTrace(candidates,target,path,0);
        return allPath;
    }

    public void backTrace(int[] candidates,int target, List<Integer> path,int start){

        if (target < 0) {
            return ;
        }
        //满足结束条件,加到总路径
        if(target==0){
            allPath.add(new ArrayList(path));
            return;
        }

        for (int i = start; i < candidates.length; i++) {
            //可选列表:当前结点大于要走的路径数值
            if (target >= candidates[i]) {
                //做选择
                path.add(candidates[i]);
                //递归
                backTrace(candidates, target - candidates[i], path, i);
                //撤销选择
                path.remove(path.size() - 1);
            }
         
        }
    }
}



相关推荐

自学Python,写一个挨打的游戏代码来初识While循环

自学Python的第11天。旋转~跳跃~,我~闭着眼!学完循环,沐浴着while的光芒,闲来无事和同事一起扯皮,我说:“编程语言好神奇,一个小小的循环,竟然在生活中也可以找到原理和例子”,同事也...

常用的 Python 工具与资源,你知道几个?

最近几年你会发现,越来越多的人开始学习Python,工欲善其事必先利其器,今天纬软小编就跟大家分享一些常用的Python工具与资源,记得收藏哦!不然下次就找不到我了。1、PycharmPychar...

一张思维导图概括Python的基本语法, 一周的学习成果都在里面了

一周总结不知不觉已经自学Python一周的时间了,这一周,从认识Python到安装Python,再到基本语法和基本数据类型,对于小白的我来说无比艰辛的,充满坎坷。最主要的是每天学习时间有限。只...

三日速成python?打工人,小心钱包,别当韭菜

随着人工智能的热度越来越高,许多非计算机专业的同学们也都纷纷投入到学习编程的道路上来。而Python,作为一种相对比较容易上手的语言,也越来越受欢迎。网络上各类网课层出不穷,各式广告令人眼花缭乱。某些...

Python自动化软件测试怎么学?路线和方法都在这里了

Python自动化测试是指使用Python编程语言和相关工具,对软件系统进行自动化测试的过程。学习Python自动化测试需要掌握以下技术:Python编程语言:学习Python自动化测试需要先掌握Py...

Python从放弃到入门:公众号历史文章爬取为例谈快速学习技能

这篇文章不谈江流所专研的营销与运营,而聊一聊技能学习之路,聊一聊Python这门最简单的编程语言该如何学习,我完成的第一个Python项目,将任意公众号的所有历史文章导出成PDF电子书。或许我这个Py...

【黑客必会】python学习计划

阅读Python文档从Python官方网站上下载并阅读Python最新版本的文档(中文版),这是学习Python的最好方式。对于每个新概念和想法,请尝试运行一些代码片段,并检查生成的输出。这将帮助您更...

公布了!2025CDA考试安排

CDA数据分析师报考流程数据分析师是指在不同行业中专门从事行业数据搜集、整理、分析依据数据作出行业研究评估的专业人员CDA证书分为1-3级,中英文双证就业面广,含金量高!!?报考条件:满18...

一文搞懂全排列、组合、子集问题(经典回溯递归)

原创公众号:【bigsai】头条号:程序员bigsai前言Hello,大家好,我是bigsai,longtimenosee!在刷题和面试过程中,我们经常遇到一些排列组合类的问题,而全排列、组合...

「西法带你学算法」一次搞定前缀和

我花了几天时间,从力扣中精选了五道相同思想的题目,来帮助大家解套,如果觉得文章对你有用,记得点赞分享,让我看到你的认可,有动力继续做下去。467.环绕字符串中唯一的子字符串[1](中等)795.区...

平均数的5种方法,你用过几种方法?

平均数,看似很简单的东西,其实里面包含着很多学问。今天,分享5种经常会用到的平均数方法。1.算术平均法用到最多的莫过于算术平均法,考试平均分、平均工资等等,都是用到这个。=AVERAGE(B2:B11...

【干货收藏】如何最简单、通俗地理解决策树分类算法?

决策树(Decisiontree)是基于已知各种情况(特征取值)的基础上,通过构建树型决策结构来进行分析的一种方式,是常用的有监督的分类算法。决策树算法是机器学习中的一种经典算法,它通过一系列的规则...

面试必备:回溯算法详解

我们刷leetcode的时候,经常会遇到回溯算法类型题目。回溯算法是五大基本算法之一,一般大厂也喜欢问。今天跟大家一起来学习回溯算法的套路,文章如果有不正确的地方,欢迎大家指出哈,感谢感谢~什么是回溯...

「机器学习」决策树——ID3、C4.5、CART(非常详细)

决策树是一个非常常见并且优秀的机器学习算法,它易于理解、可解释性强,其可作为分类算法,也可用于回归模型。本文将分三篇介绍决策树,第一篇介绍基本树(包括ID3、C4.5、CART),第二篇介绍Ran...

大话AI算法: 决策树

所谓的决策树算法,通俗的说就是建立一个树形的结构,通过这个结构去一层一层的筛选判断问题是否好坏的算法。比如判断一个西瓜是否好瓜,有20条西瓜的样本提供给你,让你根据这20条(通过机器学习)建立起...