【算法题】46. 全排列-力扣(LeetCode)
【算法题】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代码
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.结语
本人资历尚浅,发博客主要是记录与学习,欢迎大佬们批评指教!大家也可以在评论区多多交流,相互学习,共同成长。