题目如下
给定一个可包含重复数字的序列,返回所有不重复的全排列。
示例:
输入: [1,1,2]
输出:
[
[1,1,2],
[1,2,1],
[2,1,1]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/permutations-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
代码如下
class Solution:
def permuteUnique(self, nums: List[int]) -> List[List[int]]:
res = []
n = len(nums)
nums.sort()
#global visit = [[] for i in range (n)]
def back(i,tmp,nums):
if i == n:
#if tmp not in res:
res.append(tmp)
return
# if tmp in res:
# return
for j in range(len(nums)):
if j > 0 and nums[j] == nums[j-1]:
continue
back(i+1,tmp+[nums[j]],nums[:j]+nums[j+1:])
return
back(0,[],nums)
return res
- 主要采用的思想是回溯和剪枝
- res存放返回的全排列列表
- tmp存放一个排列
- nums存放向下递归的数字序列
- 在没有使用剪枝的时候,时间为2000多ms
剪枝的代码为:
if j > 0 and nums[j] == nums[j-1]:
continue
- 这个剪枝需要对nums先排序
- 当 nums[j] == nums[j-1] 的时候,就可以跳过nums[j]往下的递归,达到剪枝的目的
- 剪枝之后,运行的时间为84ms