给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
输入: [1,2,2]
输出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subsets-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路一
将排序后的nums数组的全排列插入到set之中,利用stl的set进行去重,可以预见的是,排序后的全排列不会产生因为顺序不同而产生的相同子集,因为子集也全都是升序的,因此可以利用set去重,效率较低,但直观
class Solution {
public:
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
int len = nums.size();
sort(nums.begin(),nums.end());
set<vector<int>> ret;
int retlen = pow(2,len);
vector<vector<int>> result;
for(int i = 0;i<retlen;i++)
{
vector<int> tvector;
int temp = i;
for(int j = len-1;j>=0;j--)
{
if((temp&1)==1)
{
tvector.push_back(nums[j]);
}
temp = temp>>1;
}
ret.insert(tvector);
}
set<vector<int>>::iterator it;
for(it = ret.begin();it!=ret.end();it++)
{
result.push_back(*it);
}
return result;
}
};
思路二
利用回溯剪枝来去掉重复,同样的,对数组进行排序来排除,排列顺序不同但元素相同的重复。之后在回溯的过程中,先假设对每个位置都进行完选择再将数组push到结果数组之中,因此先选择所有元素都push到数组之中,然后进行回溯,回溯之后并不马上继续向下遍历,而是进行判断,判断当前弹出的元素(回溯所弹出的元素)是否与其之后的元素相同,若相同则表明这个元素也不应该被选入,因此跳过,这样就可以避免因为不同位置元素值相同,而导致的重复。
class Solution {
public:
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
int len = nums.size();
sort(nums.begin(),nums.end());
vector<int> sub;
vector<vector<int>> ret;
generate(0,nums,sub,ret);
return ret;
}
void generate(int cur,vector<int> &nums,vector<int> &sub,vector<vector<int>> &ret)
{
if(cur==nums.size())
{
ret.push_back(sub);
return;
}
sub.push_back(nums[cur]);
generate(cur+1,nums,sub,ret);
sub.pop_back();
while(cur+1<nums.size()&&nums[cur]==nums[cur+1])
{
cur++;
}
generate(cur+1,nums,sub,ret);
}
};