90. 子集 II
给你一个整数数组 nums
,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。
示例 1:
输入:nums = [1,2,2]
输出:[[],[1],[1,2],[1,2,2],[2],[2,2]]
示例 2:
输入:nums = [0]
输出:[[],[0]]
本题解基于「78. 子集的官方题解」,请读者在充分理解该题解后继续阅读。
方法一:迭代法实现子集枚举
思路
考虑数组 [1,2,2],选择前两个数,或者第一、三个数,都会得到相同的子集。
也就是说,对于当前选择的数 x,若前面有与其相同的数 y,且没有选择 y,此时包含 x 的子集,必然会出现在包含 y 的所有子集中。
我们可以通过判断这种情况,来避免生成重复的子集。代码实现时,可以先将数组排序;迭代时,若发现没有选择上一个数,且当前数字与上一个数相同,则可以跳过当前生成的子集。
class Solution {
List<Integer> t = new ArrayList<Integer>();
List<List<Integer>> ans = new ArrayList<List<Integer>>();
public List<List<Integer>> subsetsWithDup(int[] nums) {
Arrays.sort(nums);
int n = nums.length;
//1 << n 1左移n位确定循环的次数
for (int mask = 0; mask < (1 << n); ++mask) {
//每次刷新t
t.clear();
//flag判断是否前面有与其相同的数 y,且没有选择 y 为false时为存在该情况
boolean flag = true;
for (int i = 0; i < n; ++i) {
//比较mask的二进制的每一位是否为1,是1就加入相应位置的数
if ((mask & (1 << i)) != 0) {
//判断是否前面有与其相同的数 y,且没有选择 y
if (i > 0 && (mask >> (i - 1) & 1) == 0 && nums[i] == nums[i - 1]) {
flag = false;
break;
}
t.add(nums[i]);
}
}
//将t加入集合
if (flag) {
ans.add(new ArrayList<Integer>(t));
}
}
return ans;
}
}
方法二:递归法实现子集枚举
思路
与方法一类似,在递归时,若发现没有选择上一个数,且当前数字与上一个数相同,则可以跳过当前生成的子集。
class Solution {
List<Integer> t = new ArrayList<Integer>();
List<List<Integer>> ans = new ArrayList<List<Integer>>();
public List<List<Integer>> subsetsWithDup(int[] nums) {
Arrays.sort(nums);
//开始递归,从位置0开始
dfs(false, 0, nums);
return ans;
}
public void dfs(boolean choosePre, int cur, int[] nums) {
if (cur == nums.length) {
// 记录答案
ans.add(new ArrayList<Integer>(t));
return;
}
// 不考虑选择当前位置
dfs(false, cur + 1, nums);
if (!choosePre && cur > 0 && nums[cur - 1] == nums[cur]) {
return;
}
t.add(nums[cur]);
// 考虑选择当前位置
dfs(true, cur + 1, nums);
t.remove(t.size() - 1);
}
}