90. 子集 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
题解
这道题难就难在去重,我们如何把在搜索结果集的时候去重呢。
例如给出的示例1,第二第三个元素重复了,当我们枚举了一二元素(即[1,2]),然后枚举一三元素时(即[1,2]),这就有相同子集了。
再举个例子[1,2,2,2],当我们枚举了一二三元素(即[1,2,2]),然后枚举一三四元素时(即[1,2,2]),又出现相同子集了。
通过上面两个例子知道,当我们前面出现一个相同的元素(例如2),我们没有选择,此时选择当前元素(2),就会出现重复的子集。因为如果选择了前一个元素,和另一种不选择前一个元素,选择当前元素,都是同样的子集。所以我们要处理这种情况,判断前一个元素是否选取,然后和前一个元素是否相同。
class Solution {
List<List<Integer>> ans = new ArrayList<List<Integer>>();//结果集
List<Integer> queue = new ArrayList<Integer>();//辅助队列
public List<List<Integer>> subsetsWithDup(int[] nums) {
Arrays.sort(nums);//排序,把相同元素都整理在一起
dfs(false, 0, nums);//没有选取任何元素,从0开始选取
return ans;
}
void dfs(boolean choosePre, int cur, int[] nums){
if(cur == nums.length){//到达数组末尾
ans.add(new ArrayList<Integer>(queue));
return;
}
dfs(false, cur + 1, nums);//没有选取当前元素,枚举下一个元素
if(!choosePre && cur > 0 && nums[cur - 1] == nums[cur]){//如果没有选取前一个元素,当前元素和前一个元素相同
return;
}
queue.add(nums[cur]);//选取当前元素
dfs(true, cur + 1, nums);//枚举下一个元素
queue.remove(queue.size() - 1);//去除当前元素
}
}