LeetCode 90. 子集 II

Description

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

说明：解集不能包含重复的子集。

示例:

输入: [1,2,2]
输出:
[
  [2],
  [1],
  [1,2,2],
  [2,2],
  [1,2],
  []
]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subsets-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

Solution

注意相对于子集问题而言，这里由于会出现重复元素，需要先排序保证回溯出的track是按照字典顺序出现，就不会出现[2,1,2]和[2,2,1]等track顺序不同的情况。

class Solution:
    def subsetsWithDup(self, nums: List[int]) -> List[List[int]]:
        res = []
        nums.sort() # 注意先排序
        def backtrack(nums, track):
            if track not in res:
                res.append(track[:])
            for i in range(len(nums)):
                track.append(nums[i])
                backtrack(nums[i+1:], track)
                track.pop()
        backtrack(nums, [])
        return res