LeetCode 90. 子集 II（C++、python）

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

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

示例:

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

C++

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

python

class Solution:
    def DFS(self,res,tmp,nums,k):
        if tmp not in res:
            res.append(tmp.copy())
        if k==len(nums):
            return
        for i in range(k,len(nums)):
            tmp.append(nums[i])
            self.DFS(res,tmp,nums,i+1)
            del tmp[-1]
    
    def subsetsWithDup(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums.sort()
        res=[]
        tmp=[]
        self.DFS(res,tmp,nums,0)
        return res