[算法题解详细]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;
    }
};

猜你喜欢

转载自blog.csdn.net/m0_61607810/article/details/121284953