DFS+回溯专题 - leetcode78. Subsets/90. Subsets II

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/a786150017/article/details/82961935

78. Subsets

题目描述

没有重复数字的全组合

例子

Input: nums = [1,2,3]
Output:
[
[3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[]
]

解法
法1 - 常规DFS
法2 - 非递归，可以看出是在上一次结果的基础上加入新元素

解法1
递归DFS

class Solution(object):
    def subsets(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        res = [[]]
        self.dfs(nums, [], res)
        return res
    
    def dfs(self, nums, temp, res):
        if not nums:
            return
        for i in range(len(nums)):
            temp.append(nums[i])
            res.append(temp[:])
            self.dfs(nums[i+1:], temp, res)
            temp.pop()

解法2
非递归

class Solution(object):
    def subsets(self, nums):
        res = [[]]
        for num in nums:
            res += [temp + [num] for temp in res]
        return res

90. Subsets II

题目描述

包含重复数字的整数集-全组合

例子

Input: [1,2,2]
Output:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]

扫描二维码关注公众号，回复： 3505987 查看本文章

解法
法1 - DFS，在上题基础上将nums排序。然后判断当前元素nums[i]是否和nums[i-1]相等，若相等，跳出此次循环继续下一次循环。

比如，第一次DFS到尽头后为 [], [1], [1,2], [1,2,2]；回溯时-[1,2]pop出2后，[1]尝试与最后一个2组合(重复了2==2)，continue

法2 - 非递归。类似于层次添加元素[] → [1] → [2], [1, 2] → [1,2], [2,2]…
若当前元素和上一个元素相等，则本轮 应该从[添加上一个元素]之后的位置添加该元素。

解法1
i是标明开头元素是哪个，然后依次和i以后的元素进行组合

class Solution(object):
    def subsetsWithDup(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums.sort()
        res = [[]]
        self.dfs(nums, [], res)
        return res
    
    def dfs(self, nums, temp, res):
        if not nums:
            return
        for i in range(len(nums)):
            if i > 0 and nums[i] == nums[i-1]:
                continue
            temp.append(nums[i])
            res.append(temp[:])
            self.dfs(nums[i+1:], temp, res)
            temp.pop()

精简代码

class Solution(object):
    def subsetsWithDup(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums.sort()
        res = [[]]
        self.dfs(nums, [], res)
        return res
    def dfs(self, nums, path, res):
        for i in range(len(nums)):
            if i > 0 and nums[i] == nums[i-1]:
                continue
            self.dfs(nums[i+1:], path + [nums[i]], res)
            res.append(path + [nums[i]])

解法2
非递归。若当前元素和上一个元素相等，则本轮 应该从[添加上一个元素]之后的位置添加该元素。

class Solution(object):
    def subsetsWithDup(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums.sort()
        res = [[]]
        for i in range(len(nums)):
            if i == 0 or nums[i] != nums[i-1]:
                l = len(res)
            for path in res[len(res)-l:]:
                res.append(path + [nums[i]])
        return res