题目链接:https://leetcode-cn.com/problems/subsets-ii/
题目描述
给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
**说明:**解集不能包含重复的子集。
示例:
输入: [1,2,2]
输出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]
思路
(1)先排序一遍,方便处理重复数字的情况
(2)依照 78题 子集 的做法,迭代地增加每个元素;注意要判断本次元素和上一个元素是否为同一个,如果是同一个,则只从上一轮开始增添元素的位置开始遍历,否则从头遍历
时间复杂度:O(2^n)
空间复杂度:O(1)
代码
class Solution {
public:
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
vector<vector<int>> ret = {{}};
if(nums.empty()) return ret;
sort(nums.begin(), nums.end()); // 排序
int start = 0;
for (int i = 0; i < nums.size(); ++i) {
int len = ret.size();
if (i>0 && nums[i]!=nums[i-1])
start = 0; // 新数字,全部遍历
for(int j = start; j < len; ++j){
vector<int> tmp = ret[j];
tmp.push_back(nums[i]);
ret.push_back(tmp);
}
start = ret.size() - (len - start); // 重复数字,从上一轮新添加的部分开始
}
return ret;
}
};