[算法题解详细]DFS解力扣90子集Ⅱ

题目

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

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

示例1

输入：nums = [1,2,2]
输出：[[],[1],[1,2],[1,2,2],[2],[2,2]]

示例2

输入：nums = [0]
输出：[[],[0]]

示例3

 1. 1 <= nums.length <= 10
 2. -10 <= nums[i] <= 10

思路

这道题是子集这道题的变型，子集是数组中的元素互不相同，但这道题数组中的元素是可以相同的，所以我们不仅要剪枝[1 , 2]，[2 , 1]这种的重复子集，还要剪枝两个重复数组元素，下面我画了示例1的解空间树

通过上面的示例图片可以看到剪枝的地方，下面我们看代码

代码

主函数中还是正常的初始化ans和temp容器，为了简洁dfs函数的参数，所以定义到类里，然后这里要注意我们要先将给定的nums数组进行排序，这是为了让两个相同的元素排在相邻的位置，然后进入dfs函数中

class Solution {
     
public:
    int n;
    vector<vector<int>> ans;
    vector<int> temp;
    vector<vector<int>> subsets(vector<int>& nums) {
     
        n = nums.size();
        dfs(n , nums, 0);
        return ans;
    }
};

然后是dfs函数中，依旧是直接把答案加入ans容器中，然后是循环，循环起点仍然是传入的参数u，这些代码都和之前那篇文章子集一样，只是今天这道题中我们要多一步判断是不是重复元素，如果是则直接剪枝

void dfs(int n, vector<int>& nums, int u) {
     
        ans.push_back(temp);
        for(int i = u; i < n; i++) {
     
            if(i > u && nums[i] == nums[i - 1]) {
     
                continue;
            }
            temp.push_back(nums[i]);
            dfs(n, nums, i + 1);
            temp.pop_back();
        }
}

i > u就代表是i之后的元素，nums[i] == nums[i - 1]判断两元素是否相等，如果满足这两个条件则说明是重复元素，直接continue剪枝
下面是完整代码：

class Solution {
     
public:
int n;
    vector<vector<int>> ans;
    vector<int> temp;
    void dfs(int n, vector<int>& nums, int u) {
     
        ans.push_back(temp);
        for(int i = u; i < n; i++) {
     
            if(i > u && nums[i] == nums[i - 1]) {
     
                continue;
            }
            temp.push_back(nums[i]);
            dfs(n, nums, i + 1);
            temp.pop_back();
        }
    }
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
     
        sort(nums.begin(), nums.end());
        n = nums.size();
        dfs(n , nums, 0);
        return ans;
    }
};