代码随想录系列文章目录
回溯篇- 排列
经历了,组合,切割字串(收集叶子结点)
子集,(收集所有结点)
现在来到了排列
46.全排列
题目链接
首先这个题是排列,比如(1,2,3)和(1,3,2)是不一样的排列,都应该算进Res中
那么画出递归树的时候,大概是这样的:
可以看出来,每层递归的逻辑应该是正常的,但是下一层,不需要从start+1开始遍历
举个例子,上一层选了3,下一层可以从头开始选1,因此排列不需要有startidx.
但是,单纯不需要start参数,那下一层不是会把自己也选中么?
所以说,这又涉及了去重,而这个去重是递归下一层时的去重,而不是在本层中的去重
在递增子序列中,涉及在本层中的去重是因为,每一层都是nums[…]的切片,这个nums本身就有重复数字,因此我们在遍历一层的时候,采用一个set去去重
而全排列的去重是在path中已经有的元素,下一层重头遍历时要避开它,因此代码可以这样写:
class Solution:
def permute(self, nums: List[int]) -> List[List[int]]:
res = []
n = len(nums)
def rec(path):
if len(path) == n:
res.append(path)
return
for i in range(n):
if nums[i] not in path:
rec(path+[nums[i]])
else:
continue
rec([])
return res
总结:全排列出口还是收集叶子结点,也就是长度和nums一样的时候收集,每层递归逻辑不需要start, 下一层从头开始但是要把已经在path里的 ‘自己’ 跳过
47.全排列 II
这个题就涉及了,每层的去重,外加每一枝上的去重
然后呢,由于在46题中,每一层中是没有重复的,我就简单用了
if nums[i] not in path: rec(path+[nums[i]])
去重,如果在这道题中是不可行的,从递归树可以看到:
最左边的一支,在上一层选了1的情况下,下一层依然选了1,但是此‘1’非彼‘1’,也就是,上一层选过的’1’, 我们不能选,未选过的’1’,我们可以选。因此,就需要一个used数组,去递归标记每一位是否用过了,这个used数组和nums一样长度,为的是让递归遍历时每一位能对的上。
思路如下图:
class Solution:
def permuteUnique(self, nums: List[int]) -> List[List[int]]:
res = []
used = [0] * (len(nums))
def rec(used, path):
if len(path) == len(nums):
res.append(path)
return
a = []
for i in range(len(nums)):
if nums[i] in a:
continue
if used[i] == 1:
continue
used[i] = 1
rec(used, path+[nums[i]])
used[i] = 0
a.append(nums[i])
rec(used, [])
return res
由于used[i]赋1的操作我们不能在参数里去做,所以必须回溯,而且段逻辑也挺巧妙的,前面两个continue必须这么写