90. Subsets II (M)

Given a collection of integers that might contain duplicates, nums, return all possible subsets (the power set).

Note: The solution set must not contain duplicate subsets.

Example:

Input: [1,2,2]
Output:
[
  [2],
  [1],
  [1,2,2],
  [2,2],
  [1,2],
  []
]

题意

求一个数组的幂集(即所有子集，包括全集和空集，构成的集族)，数组中可以有重复元素，不允许有重复的子集。

思路

与 78. Subsets 相比，数组中多了重复元素，在使用回溯法时就需要判断要不要把当前元素加入到结果集中。

对于数组[2, 2]来说，如果按照先前方法递归，会得到以下4组子集(空格表示递归时没有选取对应位置的元素)：

但可以看到这4组中，第2、3组实际上是一个集合，关键在于对第二个2是否选取的判断。

为了避免重复集合，可以采取以下措施：记 nums[index] 为当前待选取的元素，如果 nums[index] == nums[index - 1]，则当且仅当 nums[index - 1] 已经在列表中，才需要将 nums[index] 加入列表并继续递归，否则不将 nums[index] 加入列表。以数组[2, 2, 2]为例实行上面的做法：

代码实现 - 递归

class Solution {
    public List<List<Integer>> subsetsWithDup(int[] nums) {
        List<List<Integer>> ans = new ArrayList<>();
        Arrays.sort(nums);                                // 先排序便于处理
        dfs(nums, 0, new ArrayList<>(), ans, false);
        return ans;
    }
    private void dfs(int[] nums, int index, List<Integer> list, List<List<Integer>> ans, boolean preAdded) {
        if (index == nums.length) {
            ans.add(new ArrayList<>(list));
            return;
        }
        // 如果与前一个元素相同则需要进行特殊处理，否则按正常做法分两条路递归
        if (index > 0 && nums[index] == nums[index - 1]) {
            // 只有当前一个元素已经加入，才需要加入当前元素
            if (preAdded) {
                list.add(nums[index]);
                dfs(nums, index + 1, list, ans, true);
                list.remove(list.size() - 1);
            }
            dfs(nums, index + 1, list, ans, false);
        } else {
            list.add(nums[index]);
            dfs(nums, index + 1, list, ans, true);
            list.remove(list.size() - 1);
            dfs(nums, index + 1, list, ans, false);
        }
    }
}

代码实现 - 迭代

class Solution {
    public List<List<Integer>> subsetsWithDup(int[] nums) {
        List<List<Integer>> ans = new ArrayList<>();
        Arrays.sort(nums);
        ans.add(new ArrayList<>());
        int preSize = 0;        // 记录了结果集中添加了上一元素的第一个列表的下标
        for (int i = 0; i < nums.length; i++) {
            int size = ans.size();
            // 若为重复元素，则只在添加了上一元素的列表后添加当前元素
            int index = i > 0 && nums[i] == nums[i - 1] ? preSize : 0;
            for (int j = index; j < size; j++) {
                List<Integer> temp = new ArrayList<>(ans.get(j));
                temp.add(nums[i]);
                ans.add(temp);
            }
            preSize = size;
        }
        return ans;
    }
}