LeetCode 90. 子集 II（Subsets II）

题目描述

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

说明：解集不能包含重复的子集。

示例:

输入: [1,2,2]
输出:
[
  [2],
  [1],
  [1,2,2],
  [2,2],
  [1,2],
  []
]

解题思路

利用回溯的思想，对于每个元素分为两种情况：不添加到子集或者添加到子集。注意因为有重复元素，所以先对整个数组排序使得重复元素到一起，若遇到重复元素，则单独处理，针对每种可能的组合情况添加或不添加到子集中。

代码

class Solution {
public:
    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        vector<vector<int>> res;
        vector<int> temp;
        sort(nums.begin(), nums.end());
        subsets(nums, 0, temp, res);
        return res;
    }
    void subsets(vector<int> nums, int idx, vector<int> temp, vector<vector<int>> &res){
        if(idx == nums.size()){
            res.push_back(temp);
            return;
        }
        if(idx < nums.size() - 1 && nums[idx] == nums[idx + 1]){
            int i = idx;
            while(i < nums.size() - 1 && nums[i] == nums[i + 1]) i++;
            while(idx <= i){
                subsets(nums, i + 1, temp, res);
                temp.push_back(nums[idx++]);
            }
            subsets(nums, i + 1, temp, res);
        }
        else{
            subsets(nums, idx + 1, temp, res);
            temp.push_back(nums[idx]);
            subsets(nums, idx + 1, temp, res);
        }
    }
};