题目描述
给定一组可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明: 解集不能包含重复的子集。
示例:
输入: nums = [1,2,2]
输出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]
我的思路
- 是LeetCode-子集的加强版,增加了一些判断条件,整体的思路没变。
- 同样使用回溯法,属于没有递归基的递归,每调用一次递归,就输出当前 temp 中的元素。例如本题,元素弹出的顺序为 1 | 1,2 | 1,2,2 | 1,2 | 2,2 | 2 。递归树如下:
root
/ | \
1 2 2
/ \ |
2 2 2
|
3
- 但我们会发现,1,2 数组有两个,不符题意。因此每递归到一层,应该去掉该层相同的数字,比如例子中该去掉第二层的第二个2。
- 还有一点不同,需要在运算前,将原数组进行排序。
代码如下:
class Solution {
vector<vector<int>> res;
vector<int> temp;
public:
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
sort(nums.begin(), nums.end());
recur(nums, 0);
return res;
}
void recur(vector<int>& nums, int index){
res.push_back(temp);
for(int i = index; i < nums.size(); i++){
if(i > index && nums[i] == nums[i-1]) continue; //去掉该层相同的数字
temp.push_back(nums[i]);
recur(nums, i + 1);
temp.pop_back();
}
return;
}
};