Leetcode 90. 子集 II

zjshuster2024-07-01 13:12

给你一个整数数组 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,2,2]，选择前两个数，或者第一、三个数，都会得到相同的子集。

也就是说，对于当前选择的数 x，若前面有与其相同的数 y，且没有选择 y，此时包含 x 的子集，必然会出现在包含 y 的所有子集中。

我们可以通过判断这种情况，来避免生成重复的子集。代码实现时，可以先将数组排序；迭代时，若发现没有选择上一个数，且当前数字与上一个数相同，向下一个位置递归迭代时，当前递归路径下的集合不需要选取当前元数，以避免重复。

代码实现

java 复制代码

class Solution {
     public List<List<Integer>> subsetsWithDup(int[] nums) {
        Set<List<Integer>> r = new HashSet<>();
        Arrays.sort(nums);
        dfs23(nums, r, 0, new ArrayList<>());
        return new ArrayList<>(r);
    }

    private void dfs23(int[] nums, Set<List<Integer>> r, int i, List<Integer> t) {
        if (i == nums.length) {
            r.add(new ArrayList<>(t));
            return;
        }
        for (int j = i; j < nums.length; j++) {
            dfs23(nums, r, j + 1, t); // 不选取元素
            if (j > 0 && nums[j] == nums[j - 1] && (t.size() > 0 && t.get(t.size() - 1) != nums[j])) {
                continue;
            }
            t.add(nums[j]);  // 选取元素
            dfs23(nums, r, j+ 1, t);
            t.remove(t.size() - 1);
        }
    }
}