题目描述

给定一个可能包含重复元素的整数数组 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, , 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() - && nums[idx] == nums[idx + ]){
int i = idx;
while(i < nums.size() - && nums[i] == nums[i + ]) i++;
while(idx <= i){
subsets(nums, i + , temp, res);
temp.push_back(nums[idx++]);
}
subsets(nums, i + , temp, res);
}
else{
subsets(nums, idx + , temp, res);
temp.push_back(nums[idx]);
subsets(nums, idx + , temp, res);
}
}
};
05-12 21:15