子集 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

思路

本题是在子集的基础上添加了相同元素。因此直接使用子集的方法会得出相同子集,需要过滤。

对于数组[2,3,3],相同子集就是第1位、第2位组成的集合和第1位、第3位组成的集合。观察规律,

如果相邻的两个数相等,则其中一个数不在子集中,另一个数在子集中的时候,就会出现相同子集。

所以,可以针对这种情况过滤,实现上,先对数组进行排序,然后依次添加子集,如果当前位置的数

和前一个数相等,并且前一个数不在当前子集里,则认为当前子集会和之前的子集相等,则不加入答案。

代码

class Solution {
public:
    vector<vector<int>> subsetsWithDup(vector<int> &nums) {
        sort(nums.begin(), nums.end());
        int n = nums.size();
        vector<vector<int>> ans;
        for (int mask = 0; mask < (1 << n); ++mask) {
            vector<int> t;
            bool flag = true;
            for (int i = 0; i < n; ++i) {
                if (mask & (1 << i)) {
                    //当前数和前一个数相等,且前一个数不在当前子集里,需要过滤
                    if (i > 0 && (mask >> (i - 1) & 1) == 0 && nums[i] == nums[i - 1]) {
                        flag = false;
                        break;
                    }
                    t.push_back(nums[i]);
                }
            }
            if (flag) {
                ans.push_back(t);
            }
        }
        return ans;
    }
};