算法之【回溯算法】详解(python)

简介: 算法之【回溯算法】详解(python)

定义


回溯算法实际上**基于DFS(深度优先搜索)**的一个类似枚举的搜索尝试过程,主要是在搜索尝试过程中寻找问题的解,当发现已不满足求解条件时,就“回溯”返回到上一个状态,尝试其他的路径,这种走不通就退回再走的技术为回溯法;满足回溯条件的某个状态的点称为“回溯点”。


回溯相关问题


  1. DFS 和回溯算法区别


DFS 是一个劲的往某一个方向搜索,直到到达最底层,而回溯算法建立在 DFS 基础之上的,但不同的是在搜索过程中,达到结束条件后,恢复状态,回溯上一层,再次搜索。因此回溯算法与 DFS 的区别就是有无状态重置


  1. 何时使用回溯算法


当问题碰到走不通的路径,需要"回头",以此来查找出所有的解的时候,使用回溯算法。即满足结束条件或者发现不是正确路径的时候(走不通),要撤销选择,回退到上一个状态,继续尝试,直到找出所有解为止。


  1. 回溯算法的基本步骤


  • 找到状态变量(回溯函数的参数)
  • 依据题意定义递归结束条件
  • 找准选择列表(与函数参数相关),与第一步紧密关联
  • 判断是否需要剪枝,即提前将不符合条件的路径排除掉
  • 作出选择,递归调用,进入下一层
  • 撤销选择



  1. 回溯算法类的题型有哪些


  • 子集、组合类问题
  • 排列类问题
  • 搜索、N皇后类问题、


注意:子集、组合是无关顺序的,而排列是和元素顺序有关的,如 [1,2] 和 [2,1] 是同一个组合(子集),但 [1,2] 和 [2,1] 是两种不一样的排列!!!!


回溯算法的通用模板


result = []
def backtrack(路径, 选择列表):
    if 满足结束条件:
        result.add(路径)
        return
    for 选择 in 选择列表:
        做选择
        backtrack(路径, 选择列表)
        撤销选择


题目举例


Leetcode78.子集


问题描述


给定一组不含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。

**说明:**解集不能包含重复的子集。


示例


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

输出:

[

[3],

[1],

[2],

[1,2,3],

[1,3],

[2,3],

[1,2],

[]

]


python代码题解


套用上述回溯算法的模板,path表示已选择的路径,for i in range(start, len(nums))表示当前能够选择的列表元素,注意对于组合类问题不能够选择前面已经选择过的元素,因为会存在重复结果,因此必须有一个start参数来控制每一轮能够选择的元素[start, len(nums)]。


class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        def backtrack(nums, path, start):
            # 将path添加到res结果中
            res.append(path.copy())
            # 当前能够选择的参数列表
            for i in range(start, len(nums)):
                # 做选择
                path.append(nums[i])
                backtrack(nums, path, i+1)
                # 撤销选择
                path.pop()
        res = []
        backtrack(nums, [], 0)
        return res


Leetcode77.组合


问题描述


给定两个整数 nk,返回 1 … n 中所有可能的 k 个数的组合。


示例


输入: n = 4, k = 2

输出:

[

[2,4],

[3,4],

[2,3],

[1,2],

[1,3],

[1,4],

]


python代码题解


本题与上一题基本相同,都是属于组合类问题,只是递归的终止条件不同,本题的终止条件是当路径长度为k时len(track) == k,将结果添加到res中。


套用上述回溯算法的模板,track表示已选择的路径,for i in range(start, n+1)表示当前能够选择的列表元素,注意start是从1开始的,应为题目中指明能够选择的数为1...n。


class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        # 数的选择范围在1-n
        def backtrack(n,k,start,track):
            if len(track) == k:
                res.append(track.copy())
                return
            # 注意i从start开始递增
            for i in range(start, n+1):
                # 做选择
                track.append(i)
                backtrack(n,k,i+1,track)
                # 撤销选择
                track.pop()
        res = []
        track = []
        backtrack(n,k,1,track)
        return res


通过上述讲解,读者应该对回溯算法的概念以及模板套路有一个基本的认识,回溯的关键在于选择与撤销选择的过程,读者可以仔细体会一下 ,相信一定会有所收获。后续会继续更新关于回溯算法的相关题解,欢迎持续关注!

相关文章
|
9天前
|
机器学习/深度学习 算法 机器人
【机器人路径规划】基于D*算法的机器人路径规划(Python代码实现)
【机器人路径规划】基于D*算法的机器人路径规划(Python代码实现)
|
9天前
|
机器学习/深度学习 算法 机器人
【机器人路径规划】基于改进型A*算法的机器人路径规划(Python代码实现)
【机器人路径规划】基于改进型A*算法的机器人路径规划(Python代码实现)
|
9天前
|
机器学习/深度学习 编解码 算法
【机器人路径规划】基于迪杰斯特拉算法(Dijkstra)的机器人路径规划(Python代码实现)
【机器人路径规划】基于迪杰斯特拉算法(Dijkstra)的机器人路径规划(Python代码实现)
|
9天前
|
机器学习/深度学习 算法 机器人
【机器人路径规划】基于A*算法的机器人路径规划研究(Python代码实现)
【机器人路径规划】基于A*算法的机器人路径规划研究(Python代码实现)
|
9天前
|
机器学习/深度学习 算法 机器人
【机器人路径规划】基于深度优先搜索(Depth-First-Search,DFS)算法的机器人路径规划(Python代码实现)
【机器人路径规划】基于深度优先搜索(Depth-First-Search,DFS)算法的机器人路径规划(Python代码实现)
|
9天前
|
算法 机器人 定位技术
【机器人路径规划】基于流场寻路算法(Flow Field Pathfinding)的机器人路径规划(Python代码实现)
【机器人路径规划】基于流场寻路算法(Flow Field Pathfinding)的机器人路径规划(Python代码实现)
机器学习/深度学习 算法 自动驾驶
107 0
|
17天前
|
算法 定位技术 调度
基于蚂蚁优化算法的柔性车间调度研究(Python代码实现)
基于蚂蚁优化算法的柔性车间调度研究(Python代码实现)
|
17天前
|
机器学习/深度学习 算法 PyTorch
【DQN实现避障控制】使用Pytorch框架搭建神经网络,基于DQN算法、优先级采样的DQN算法、DQN + 人工势场实现避障控制研究(Matlab、Python实现)
【DQN实现避障控制】使用Pytorch框架搭建神经网络,基于DQN算法、优先级采样的DQN算法、DQN + 人工势场实现避障控制研究(Matlab、Python实现)
|
24天前
|
机器学习/深度学习 资源调度 算法
一种多尺度协同变异的粒子群优化算法(Python代码实现)
一种多尺度协同变异的粒子群优化算法(Python代码实现)

热门文章

最新文章

推荐镜像

更多