90. 子集 II

题目描述

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

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

 

示例

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

 

思路

回溯法，用set避免重复的情况。

比如[1 4 1 4 4]。 除了【1 4 4】 和 【1 4 4】这种每一位上都相同的重复以外，【4 1 4】 【1 4 4】也属于重复情况。所以对原序列排序，【1 1 4 4 4】，拍完序后就不会出现第二种重复情况了。

 

代码

class Solution {
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        vector<int> item;
        vector<vector<int>> res;
        set<vector<int>> mySet;
        res.push_back(item);
        sort(nums.begin(), nums.end());
        generate(0, nums, item, mySet, res);
        return res;
    }
    
    void generate(int i, vector<int>& nums, vector<int>& item, set<vector<int>>& mySet, vector<vector<int>>& res){
        if (i == nums.size())
            return;
        item.push_back(nums[i]);
        if (mySet.count(item) == 0){
            mySet.insert(item);
            res.push_back(item);
        }
        generate(i+1, nums, item, mySet, res);
        item.pop_back();
        generate(i+1, nums, item, mySet, res);
    }
};