Python的回溯算法

什么时候用

回溯算法也叫试探法，它是一种系统地搜索问题的解的方法。
用回溯算法解决问题的一般步骤：
1、 针对所给问题，定义问题的解空间，它至少包含问题的一个（最优）解。
2 、确定易于搜索的解空间结构,使得能用回溯法方便地搜索整个解空间 。
3 、以深度优先的方式搜索解空间，并且在搜索过程中用剪枝函数避免无效搜索。
适用的问题：
子集、组合、排列、棋盘。

怎么用

模板的话长这样

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

举个例子的话

class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        ans, path, n = [], [], len(nums)
        def trackback(cur):
            ans.append(path[:])
            for i in range(cur, n):
                path.append(nums[i])
                trackback(i+1)
                path.pop()
        trackback(0)
        return ans

回溯算法解题的代码都大同小异，记住模板就够啦，需要剪枝的话可以先打印出path，观察有哪些情况是重复的需要剪掉，以上。