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()
posted @ 2021-04-02 23:21  swordspoet  阅读(26)  评论(0编辑  收藏  举报