题目

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

题解

public List<List<Integer>> subsetsWithDup(int[] nums) {
    Set<List<Integer>> result = new HashSet<>();
    int length = nums.length;
    // 空集
    result.add(new ArrayList<>());
    if (length > 0) {
        // 全集
        result.add(Arrays.stream(nums).boxed().collect(Collectors.toList()));
    }

    BiConsumer<Integer, List<Integer>> backtrack = new BiConsumer<Integer, List<Integer>>() {
        @Override
        public void accept(Integer len, List<Integer> list) {
            // 满足子集个数结束
            if (list.size() == len) {
                // 排序保证唯一性
                result.add(list.stream().map(it -> nums[it]).sorted().collect(Collectors.toList()));
                return;
            }

            for (int i = list.isEmpty() ? 0 : list.get(list.size() - 1) + 1; i < length; i++) {
                // 根据索引判断是否已经使用该元素
                // 保证按照索引递增
                list.add(i);
                this.accept(len, list);
                // 回溯
                list.remove(list.size() - 1);
            }
        }
    };

    // 子集元素个数依次回溯计算
    for (int i = 1; i < length; i++) {
        backtrack.accept(i, new ArrayList<>());
    }

    return new ArrayList<>(result);
}