LeetCode 90. 子集 II

90. 子集 II

Difficulty: 中等

示例 1： 输入：nums = [1,2,2] 输出：[[],[1],[1,2],[1,2,2],[2],[2,2]] 示例 2： 输入：nums = [0] 输出：[[],[0]]   提示： 1 <= nums.length <= 10 -10 <= nums[i] <= 10 **

给你一个整数数组 nums ，其中可能包含重复元素，请你返回该数组所有可能的子集（幂集）。

解集 不能 包含重复的子集。返回的解集中，子集可以按 任意顺序 排列。

示例 1：

输入：nums = [1,2,2]
输出：[[],[1],[1,2],[1,2,2],[2],[2,2]]

示例 2：

输入：nums = [0]
输出：[[],[0]]

提示：

• 1 <= nums.length <= 10
• -10 <= nums[i] <= 10

Solution

这题跟LeetCode 78. 子集 - swordspoet - 博客园是同一类型的题目，不过在本题中包含了重复元素，所以我们在进行组合时需要事先对数组排序，然后在回溯递归的时候注意略过数组中重复的部分，一下是递归回溯的过程。

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