給定一個可能包含重複元素的整數數組 nums,返回該數組所有可能的子集(冪集)。
說明:解集不能包含重複的子集。
示例:
輸入: [1,2,2]
輸出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]
思路:因爲題目要求子集不能重複,所以優先想到使用set。
由於set可能將[1,2]和[2,1]判斷爲不同的子集,就要先對原數組進行排序。
#include <algorithm>
class Solution {
public:
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
vector<vector<int>> result;
vector<int> item;
set<vector<int>> s;
sort(nums.begin(),nums.end());
result.push_back(item);
generate(0,result,nums,item,s);
return result;
}
private:
void generate(int i,vector<vector<int>> &result,vector<int> &nums,vector<int> &item, set<vector<int>> &s){
if(i>=nums.size()){
return;
}
item.push_back(nums[i]);
if(s.find(item)==s.end()){
result.push_back(item);
s.insert(item);
}
generate(i+1,result,nums,item,s);
item.pop_back();
generate(i+1,result,nums,item,s);
}
};