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

最新推荐文章于 2022-06-27 11:39:20 发布

原创最新推荐文章于 2022-06-27 11:39:20 发布 · 305 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 同时被 2 个专栏收录

713 篇文章

订阅专栏

回溯

25 篇文章

订阅专栏

本文探讨了在给定可能包含重复元素的整数数组情况下，如何生成所有不重复的子集（幂集）。通过使用C++和Python两种语言实现深度优先搜索(DFS)算法，确保了解集中不会出现重复的子集。文章详细介绍了算法的具体实现过程。

给定一个可能包含重复元素的整数数组 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