回溯算法——LeetCode刷题——【46. 全排列】

时代&信念

已于 2022-12-12 22:23:11 修改

阅读量227

点赞数 1

分类专栏： LeetCode 文章标签： leetcode 深度优先算法

于 2022-12-10 20:16:58 首次发布

本文链接：https://blog.csdn.net/Elon15/article/details/128268574

版权

LeetCode 专栏收录该内容

30 篇文章 2 订阅

订阅专栏

题目描述：

在这里插入图片描述

思路：

递归+回溯

以[1,2,3]举例，详细思路见下图
在这里插入图片描述

注意点：

（1）横向：遍历
（2）纵向：递归，递归深度为全排列数组长度
（3）已经出现过在used数组中的元素，直接进行剪枝
（4）当达到递归结束条件时，进行回溯，回到上一层

代码和注释：

class Solution:
    def permute(self, nums: List[int]) -> List[List[int]]:
        res = []   # 存放符合条件结果的集合
        path = []  # 用来存放符合条件的结果
        used = []  # 用来存放已经用过的数字
        def backtrack(nums,used):
            # 递归结束条件
            if len(path) == len(nums):
                return res.append(path[:])  # 此时说明找到了一组
            for i in range(0,len(nums)):
                # used里已经收录的元素，直接跳过
                if nums[i] in used:
                    continue  
                path.append(nums[i])
                used.append(nums[i])
                # 进行递归
                backtrack(nums,used)
                # 开始回溯，回到树的上一层节点
                used.pop()
                path.pop()
        # 调用函数
        backtrack(nums,used)
        # 返回结果
        return res