问题:
题目来源:力扣(LeetCode)
难度:中等
分析:
回溯法+剪枝
相比于leetcode46.全排列,题目增加了重复元素的要求。在leetcode46.全排列的基础上加两行剪枝排除重复元素就可以。
为了排除重复元素,先进行一下排序。
解决方法:
1:回溯+剪枝
class Solution:
def permuteUnique(self, nums: List[int]) -> List[List[int]]:
if len(nums) == 0: return []
def backtrack(depth):
if depth == len(nums):
res.append(path[:])
for i in range(len(nums)):
if visited[i] == True:
continue
if i > 0 and nums[i] == nums[i - 1] and visited[i - 1] == False:
continue
visited[i] = True
path.append(nums[i])
backtrack(depth + 1)
visited[i] = False
path.pop()
nums.sort()
res = []
path = []
visited = [False for i in range(len(nums))]
backtrack(0)
return res
在本题中增加的主要是这两个剪枝。
if visited[i] == True:
continue
if i > 0 and nums[i] == nums[i - 1] and visited[i - 1] == False:
continue
首先是如果备选元素已经被使用过,那么这个备选元素不能被使用第二次,这个和全排列一样。
然后要做的是排除重复元素的剪枝,i > 0是为了防止i - 1越界的,作为数组第一个元素,也不存在和数组更前面的元素相同的情况。
当元素与前面元素相同并且前面元素没有被选择时,那么这个元素也不被选择。因为如果选择这个选择元素i - 1再选i和选择i再选i - 1是重复情况,直接剪枝即可。
这里将False改为True也可以运行通过,但是速度会变慢一点。因为剪枝方式发生了改变。剪枝方式如图,取自leetcode题解。
链接:https://leetcode-cn.com/problems/permutations-ii/solution/hot-100-47quan-pai-lie-ii-python3-hui-su-kao-lu-zh/
个人感觉第一种情况,剪大枝更快更容易理解。重复元素只有前面的元素被使用了后面的元素才有被使用的机会,这样隐式地在重复元素间增加了一个优先级,使得先选择后面元素再选择前面元素的重复情况被排除。
第二种情况剪小枝是因为符合条件的元素加起来长度不够全排列被剪掉了,符合条件的重复元素的选择是从后向前选的,这样就可以保证在选择元素的时候,前一个元素没有被选择。