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/