leetcode-90-子集 II

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);
    }
}