《Leetcode of September 》回溯算法之:47. 全排列II

《Leetcode of September 》回溯算法之:47. 全排列II_第1张图片

 

看题解看到一个比较形象的图,来自https://leetcode-cn.com/problems/permutations-ii/solution/hui-su-suan-fa-python-dai-ma-java-dai-ma-by-liwe-2/ 

《Leetcode of September 》回溯算法之:47. 全排列II_第2张图片

class Solution:
    def permuteUnique(self, nums: List[int]) -> List[List[int]]:
        def huisu(n,temp,visited):
            if len(temp)==n:
                res.append(temp[:])
            
            for i in range(n):
                #如果已经用过或者和上一个是一毛一样的就进行剪枝
                #对于第二个条件,这个元素和上个元素是一样,且上个元素已经使用过,并且执行撤销操作了,所以就存在重复的元素
                if visited[i] or (i>0 and not visited[i-1] and nums[i]==nums[i-1]):
                    continue
                temp.append(nums[i])
                visited[i]=True
                huisu(n,temp,visited)
                visited[i]=False
                temp.pop()
        
        res=[]
        n=len(nums)
        #先进行排序,使得重复的数字都是相邻的
        nums.sort()
        visited = [False for _ in range(n)]
        huisu(n,[],visited)
        return res

总结:和46题差不多,但是就是有重复元素,所以就需要进行剪枝,把提前存在重复的元素剪掉。两个重要的步骤:①排序,让相邻的元素放在一起方便剪枝;②剪枝的条件判断必须和上个元素相同而且未被访问过。

你可能感兴趣的:(Leetcode,回溯,剪枝,数据结构,算法,leetcode,面试)