LeetCode 刷题系列 -- 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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/subsets-ii
 

思路：回溯算法秒杀所有排列/组合/子集问题 :: labuladong的算法小抄

回溯算法。此类型为 元素可重不可复选。
1. 先将数组中的元素排序
2. 定义数组  used ，长度为  nums.length , used[i] 代表 nums[i] 是否被放入子集中
3. 回溯算法遍历数组。注意剪枝问题。

java代码：

class Solution {
 public List<List<Integer>> subsetsWithDup(int[] nums) {
        List<List<Integer>> result = new LinkedList<>();
        boolean[] used = new boolean[nums.length];
        
        Arrays.sort(nums);
        backtrack(nums,0,used,new LinkedList<>(),result);
        
        return result;
    }

    public void backtrack(int[] nums, int start,boolean[] used, List<Integer> subSet, List<List<Integer>> result) {

        result.add(new LinkedList<>(subSet));
        
        for (int i = start; i < nums.length; i++) {
            
 // 剪枝, 若是当前元素与前一个元素相等，且前一个元素没使用过，那说明需要剪枝，因为此时进入到了新的周期，一个与之前相等元素相同的周期。
            // 具体剪枝可以看二叉树            if(i>0&&nums[i-1] == nums[i] && !used[i-1]) {
                continue;
            }
            
            subSet.add(nums[i]);
            used[i] = true;
            backtrack(nums, i + 1,used, subSet, result);
            subSet.remove(subSet.size() - 1);
            used[i] = false;
        }

    }
}