90. 子集 II

90. 子集 II

原始题目链接:https://leetcode.cn/problems/subsets-ii/

给你一个整数数组 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

解题思路:

子集问题用回溯的方法来解决,回溯法必然要用到递归。递归想好结束的条件及递归函数的定义,参数。如果不回溯,就会引起重复的问题。

代码实现:

class Solution:
    def subsetsWithDup(self, nums: List[int]) -> List[List[int]]:
        res = []
        path = []

        def back_trace(nums, index):
            res.append(path.copy())

            for i in range(index, len(nums)):
                #去重
                if i > index and nums[i] == nums[i - 1]:
                    continue
                
                path.append(nums[i])
                # 递归
                back_trace(nums, i + 1)
                # 回溯
                path.pop()
        # 去重要使用排序后的nums,这样才会根据相邻的元素是否一样来去重
        back_trace(sorted(nums), 0)
        return res

参考文献:
https://leetcode.cn/problems/subsets-ii/solution/90-zi-ji-iiche-di-li-jie-zi-ji-wen-ti-ru-djmf/