1.最开始用的方法,在上一题的基础之上加入了最后的去重判断,还加入了如果当前数和要换的数一样,就不更换的剪枝。但这个方法没有考虑到当前数和两个相同的数更换时的剪枝,所以耗时还是有点长。
class Solution:
def permuteUnique(self, nums: List[int]) -> List[List[int]]:
def BackTrack(first = 0):
if first == l:
if nums not in result:
result.append(nums[:])
for i in range(first, l):
if i == first or nums[i] != nums[first]:
nums[i], nums[first] = nums[first], nums[i]
BackTrack(first+1)
nums[i], nums[first] = nums[first], nums[i]
l = len(nums)
result = []
BackTrack()
return result
2.之后加入了上一种方法中没有的剪枝判断。而且这个剪枝步骤更简洁,只要当前数和上一个数相同,那么就直接剪枝。
import copy
class Solution:
def permuteUnique(self, nums: List[int]) -> List[List[int]]:
def BackTrack(nums, tmp = []):
if not nums:
result.append(tmp[:])
return
for i in range(len(nums)):
if i != 0 and nums[i] == nums[i-1]:
continue
BackTrack(nums[:i]+nums[i+1:], tmp+[nums[i]])
l = len(nums)
nums.sort()
result = []
BackTrack(nums)
return result