topic90:子集
题目描述:
给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。
示例:
1.输入:nums = [1,2,2]
输出:[[],[1],[1,2],[1,2,2],[2],[2,2]]
2.输入:nums = [0]
输出:[[],[0]]
思路:代码的时间复杂度为 O(2^n * nlogn)
-
计算给定整数数组的长度,并初始化结果数组为空列表 result。
-
使用 for 循环枚举子集元素个数(从 0 到数组长度),其中 i 表示子集元素个数。
-
使用 itertools 库中的 combinations() 函数生成所有可能的由 i 个元素组成的组合。
-
将每个组合添加到结果数组 result 中。
-
对结果数组 result 进行去重。这里使用了一个技巧,即将每个子集按照元素大小排序后,再将其转换为元组,然后将所有元组放入一个集合中去重。
-
将去重后的元组转换为列表,返回去重后的结果数组。
class Solution:
def subsetsWithDup(self, nums: List[int]) -> List[List[int]]:
length=len(nums)
result=[] # 初始化结果数组
# 枚举子集元素个数
for i in range(length+1):
# 生成所有可能的组合
c=itertools.combinations(nums,i)
# 将组合添加到结果数组中
for j in c:
result.append(list(j))
# 对结果数组进行去重
new_list = list(set(tuple(sorted(x)) for x in result))
new_list = [list(x) for x in new_list]
# 返回去重后的结果数组
return new_list