LeetCode-090-子集II

给定一个可能包含重复元素的整数数组 nums，返回该数组所有可能的子集（幂集）。

说明：解集不能包含重复的子集。

示例:

输入: [1,2,2]
输出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]

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

解题思路

采用深度优先搜索
dfs(int[] nums, int start, Deque<Integer> path, List<List<Integer>> result)
参数依次是数组，起始拼接位置，搜索过程中的路径，结果集
因为不能出现重复元素，比如[1, 2, 2]这个集合，会出现两个[1, 2]
这时要考虑"剪枝"，剪枝的前提是将数组排序
然后考虑剪枝的条件"同一层不能用相同数字"
比如[1, 2, 2]深度优先遍历时

image.png

代码

class Solution {
    public List<List<Integer>> subsetsWithDup(int[] nums) {
        List<List<Integer>> result = new LinkedList<>();
        if (nums == null || nums.length == 0) {
            return result;
        }
        result.add(new ArrayList<>()); // 空集
        Arrays.sort(nums);
        Deque<Integer> path = new LinkedList<>();
        dfs(nums, 0, path, result);
        return result;
    }

    private void dfs(int[] nums, int start, Deque<Integer> path, List<List<Integer>> result) {
        for (int i = start; i < nums.length; i++) {
            if (i == start || nums[i] != nums[i - 1]) {
                path.addLast(nums[i]);
                result.add(new ArrayList<>(path));
                dfs(nums, i + 1, path, result);
                path.removeLast();
            }
        }
    }
}