LeetCode Subsets II

LeetCode解题之Subsets II

原题

罗列出一个包含重复数字的集合的所有的子集。

注意点：

• 子集中的元素需要按照不降序排列
• 结果集中不能重复

例子：

输入: nums = [1,2,2]

输出:

[
  [2],
  [1],
  [1,2,2],
  [2,2],
  [1,2],
  []
]

解题思路

在 Subsets 迭代版本的基础上思考。在迭代重复元素的时候会生成重复的结果，那么在迭代重复元素时要特殊处理一下。就拿[1,2,2]来说，在迭代完1之后结果集为[[], [1]]，迭代第一个2后，[[], [1], [2], [1,2]]，接下来就要迭代重复的元素2了，此时如果遍历在迭代第一个2之前就存在的结果集元素（[[], [1]]）时，就会产生重复，我们只能在上一轮迭代产生的新的结果中继续添加。所以要一个额外的变量来表示在结果集中的哪个位置开始遍历。

AC源码

class Solution(object):
    def subsetsWithDup(self, nums):
        """ :type nums: List[int] :rtype: List[List[int]] """
        result = [[]]
        nums.sort()
        temp_size = 0
        for i in range(len(nums)):
            start = temp_size if i >= 1 and nums[i] == nums[i - 1] else 0
            temp_size = len(result)
            for j in range(start, temp_size):
                result.append(result[j] + [nums[i]])
        return result


if __name__ == "__main__":
    assert Solution().subsetsWithDup([1, 2, 2]) == [[], [1], [2], [1, 2], [2, 2], [1, 2, 2]]

欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源码。