原题地址:https://leetcode-cn.com/problems/subsets-ii/submissions/

题目描述:

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

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

示例:

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

解题方案:

这道题刚开始不知道怎么跟自己过不去非要用动态规划写...规则是先把i-1的子集全部加入,然后再加入i-1的子集中每个集合后面添加一个nums[i],但是动态规划去重太麻烦了,如果有重复的数字,除了把i-1的子集全部加入,如果nums[i]前面有k个数字于它相同,那么还要向i-1子集中含有前k个重复数字的集合中添加num[i]再加入i的子集,自己还是没想出来解法。

这是参考:https://blog.csdn.net/feliciafay/article/details/18977055

用回溯写去重就比较简单了,重复数字中只有第一个要考虑它已经包含了最多数字了,所以后面的都不需要考虑。

代码:

class Solution {
public:
    vector<vector<int>> result;
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        vector<int> temp;
        result.push_back(temp);
        getAns(0, nums, result, temp);
        return result;
    }
    void getAns(int start, vector<int>& nums, vector<vector<int>> &result, vector<int> temp)
    {
        if(start == nums.size() - 1)
        {
            temp.push_back(nums[start]);
            result.push_back(temp);
        }
            
        else
        {
            for(int i = start; i < nums.size(); i ++)
            {
                while(i != 0 && i != start && nums[i] == nums[i - 1])
                {
                    i ++;
                }
                if(i == nums.size())
                    break;
                temp.push_back(nums[i]);
                result.push_back(temp);
                getAns(i + 1, nums, result, temp);
                temp.pop_back();
            }
        }
    }
};

 

收藏 打印