一、题目描述
给你一个整数数组 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]]:
length =len(nums)
result = []
path = []
nums.sort()
for i in range(length+1):
print(f"count: {i}")
path = list(itertools.combinations(nums,i))
path = [list(path[i]) for i in range(len(path))]
print("Path:",path)
for i in range(len(path)):
if path[i] not in result:
result.append(path[i])
print(result)
return result
三、解题思路
本题是在 78. 子集 Python的基础上添加了“可能包含重复元素”这一条件,如果继续采用原本的算法,则会导致每个重复的元素都会被当成一个单独的个体被添加进入组合中,所以本题解思路为在 78. 子集 Python的基础上添加一个对重复子集的判断操作。
方法也很简单,只需要result
在每一次添加时,加入判重操作即可;
本题解中path
相当于当前子集元素的长度count
下所有的可能子集
(例如:以nums=[1,2,2]
为例,
当count=0
时,path
数组中元素的长度为0
,只有[]
一种情况;
当count=1
时,path
数组中元素的长度为1
,子集有3
种可能,为[1]、[2]、[2]
,注意此时的path
数组中是可能包含重复的子集…),
在得到当前的path
数组后,将不在result
中的元素添加进入result
即可。
之后重复多次将所有长度的子集元素都添加进入result
中即可,重复次数为nums
的长度+1次。