【算法题】46. 全排列-力扣(LeetCode)

杰九2024-09-20 23:40

【算法题】46. 全排列-力扣(LeetCode)

1.题目

下方是力扣官方题目的地址

46. 全排列

给定一个不含重复数字的数组 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]]

示例 3：

复制代码

输入：nums = [1]
输出：[[1]]

2.题解

思路

本题全排列是一个典型的深度优先搜索的问题，它很好地体现了dfs中的剪枝和回溯

首先，我们可以很容易想到深度优先，我们以nums = [1,2,3]为例，我们可以很快地构造出如下图所示的树：

不过这颗最容易想到的树有很多重复且多余的节点，我们需要将其剪掉。

如何剪掉这些枝叶呢？

我们只需要初始化一个数组，记录已经选过的数字，接下来不选已经选过的数字就行了，这就是剪枝的过程。

在进入下一层后别忘了回溯数组。

Python代码

python 复制代码

class Solution(object):
    def permute(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        global used
        ans,used=[],[]
        def dfs(d):
            global used         # 用来记录已经使用过的数，方便下面的剪枝
            if d==len(nums):
                ans.append(used)
                return
            for i in [x for x in nums if x not in used]:  # 在遍历的时候就进行剪枝
                used.append(i)
                dfs(d+1)                # 进入下一层
                used=used[:-1]          # 回溯
        dfs(0)
        return ans

3.结语

本人资历尚浅，发博客主要是记录与学习，欢迎大佬们批评指教！大家也可以在评论区多多交流，相互学习，共同成长。