淘先锋技术网

首页 1 2 3 4 5 6 7

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

解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。

示例 1:
输入:nums = [1,2,2]
输出:[[],[1],[1,2],[1,2,2],[2],[2,2]]

示例 2:
输入:nums = [0]
输出:[[],[0]]

我一开始做这道题的思路是,先对nums排序,再直接递归找到所有的组合数,然后用set去重。但显然这不是一个好方法,因为会搜索到无用的序列。

另一种想法是当回溯做弹出后,如果当前元素和下一个元素相等,就跳过深层递归,对pos 加1。比如 1 2 2 3。当下标为1的元素弹出后,因为nums[1]和nums[2]相等,所以跳过下标为2的深层递归,对下标为3做深层递归。这样做的道理是:去掉一个2的序列在第二个元素被包含时已经有了。第二个元素被弹出,第3个元素不应该再被包含,这样就直接跳过了第三个元素的递归。

class Solution {
public:
    vector<vector<int>> ret;
    vector<int> nums;
    vector<int> temp;

    void Recursion(int pos) 
    {
        if (pos == nums.size()) {
            ret.push_back(temp);
            return;
        }

        temp.push_back(nums[pos]);
        Recursion(pos + 1);
        temp.pop_back();
        while (pos + 1 < nums.size() && nums[pos + 1] == nums[pos]) {
            ++pos;
        }
        Recursion(pos + 1);
    }

    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        this -> nums = nums; 
        Recursion(0);
        
        return ret;
    }
};