90.子集 2

子集2

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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/subsets-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

同理可以先将树画出来，见注释。可以看见，下一层的开始是上一层的下一个元素，同时同一层中重复的元素不可以再次使用。需要先排序再DFS。

code

class Solution {
public:
    //1 -> 2 -> 3 -> 2
    //       -> 2
    //  -> 3

    //错误示例
    //44414
    //4 -> 4 -> 4 -> 1 -> 4
    //            -> 4
    //       -> 1 -> 4
    //4441和4414出现重复
    //具体原因不分析，排序应该是可以解决的

    vector<vector<int>> ans;
    int n;

    void dfs(vector<int>& nums,int start,vector<int> & path)
    {
        unordered_map<int,int> used;
        for(int i = start;i < n;i ++)
        {
            if(used.find(nums[i]) == used.end())
            {
                used[nums[i]] ++;
                path.push_back(nums[i]);
                ans.push_back(path);
                dfs(nums,i + 1,path);
                path.pop_back();
            }
            else continue;
        }
    }
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        n = nums.size();
        sort(nums.begin(),nums.end());
        vector<int> path;
        ans.push_back({});
        dfs(nums,0,path);
        return ans;
    }
};