90. 子集 II

90. 子集 II

难度中等809

给你一个整数数组 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

解题思路

• 这道题我们需要使用dfs+回溯的方法来进行求解。

• 由于题目明确指出列表中可能有重复数字，所以我们在dfs的时候要进行剪枝。

算法

• 将数组进行升序排列。

• 定义一个递归方法 dfs，参数有：当前子集subset，当前子集长度k，返回结果res。 将当前子集添加到res中。 从 k 到 len(nums)-1 遍历索引i。 如果i != k并且nums[i] == nums[i - 1]，说明nums[i]是重复元素，所以我们跳过nums[i]。 将nums[i] 添加到当前子集subset。 进行下一层的递归搜索，继续向子集中添加元素，这时k要加一。 从 subset中删除nums[i]进行回溯。

举例分析

• 假设nums = [1, 2, 2']，递归树如图所示。树每深一层，子集的长度就加一。每个节点都是满足条件的子集，需要记录到结果res中。

• 其中标叉的地方进行了剪枝。由于数组中有两个2，所以如果两者在同一层，只保留第一个。

 

复杂度分析

 

public class Solution {
    /**
     * @param nums: A set of numbers.
     * @return: A list of lists. All valid subsets.
     */
    public List<List<Integer>> subsetsWithDup(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        // 排序
        Arrays.sort(nums);
        // dfs搜索
        dfs(nums, 0, new ArrayList<>(), res);
        return res;
    }
    private void dfs(int[] nums, int index, List<Integer> subset, List<List<Integer>> res) {
        // 当前组合存入res
        res.add(new ArrayList<>(subset));
        // 为subset新增一位元素
        for (int i = index; i < nums.length; ++i) {
            // 剪枝 说明nums[i]是重复元素
            if (i != index && nums[i] == nums[i - 1]){
                continue;
            }
            subset.add(nums[i]);
            // 下一层搜索
            dfs(nums, i + 1, subset, res);
            // 回溯
            subset.remove(subset.size() - 1);
        }
    }
}