【回溯算法】leetcode 90. 子集 II

90. 子集 II

题目描述

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

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

示例1：

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

示例2：

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

提示

• $1<=nums.length<=10$
• $-10<=nums[i]<=10$

方法：回溯算法

解题思路

这道题目和 78.子集 区别就是集合里有重复元素了，而且求取的子集要去重。

代码

class Solution {
public:
    vector<vector<int>> result;
    vector<int> path;
    void dfs(int startIndex, vector<int>& nums, vector<bool>& used) {
        result.push_back(path);
        if(path.size() == nums.size())  return;
        for(int i = startIndex; i < nums.size(); i++) {
            if(i > 0 && nums[i] == nums[i - 1] && !used[i - 1]) continue;
            used[i] = true;
            path.push_back(nums[i]);
            dfs(i + 1, nums, used);
            used[i] = false;
            path.pop_back();
        }
    }
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        vector<bool> used(nums.size(), false);
        sort(nums.begin(), nums.end());
        dfs(0, nums, used);
        return result;
    }
};

复杂度分析

• 时间复杂度：$O(n\times 2^n)$。
• 空间复杂度：$O(n)$。