回溯 Leetcode 90 子集II

子集II

Leetcode 90

学习记录自代码随想录

给你一个整数数组 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

要点：1.和求子集基本相同，但需要进行“树层去重”；
2.注意数组不能越界，nums[i]注意不要忽略i不能=nums.size()；

class Solution {
private:
    vector<int> path;
    vector<vector<int>> result;
    
    void backtracking(vector<int>& nums, int startIndex, vector<int> used){

        result.push_back(path);
        if(path.size() == nums.size()){
            return;
        }

        for(int i = startIndex; i < nums.size(); i++){  // i <= nums.size()注意不要有等于号否则会在nums[i]越界
            // if(i > 0 && nums[i] == nums[i-1] && used[i-1] == 0){
            //     continue;
            // }
            // used[i] = 1;
            if(i > startIndex && nums[i] == nums[i-1]){  // 此种去重方式也可以
                continue;
            }
            path.push_back(nums[i]);
            backtracking(nums, i+1, used);  // 到下一层
            path.pop_back();
            // used[i] = 0;  // 注意要回溯
        }

    }
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        path.clear();
        result.clear();
        vector<int> used(nums.size(), 0);

        sort(nums.begin(), nums.end());
        backtracking(nums, 0, used);
        
        return result;
    }
};