1. 题目
给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
输入: [1,2,2]
输出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]
2. 分析
参考题目78。
这个题会出现的一个问题就是重复,例如给定输入【4,4,4,1,4】
按照之前78的程序,会同时给出【4,4,4,1】和【4,4,1,4】,然而这两个结果实际上是一个结果。出现这个问题的原因在于每一个子集是通过vector存储的,虽然从我们的观察角度这两个结果相等,但是由于vector存储,所以通过去重不能将这两个结果合二为一。
解决这个问题的措施就是在每次得到一个子集的时候,然后插入到最终的结果中。那么前面提到的两个例子【4,4,4,1】和【4,4,1,4】都会变成【1,4,4,4】,这样就能够去重了。
3. 技术记录
对vector进行排序
sort(vec.begin(),vec.end());
对vector进行去重
sort(vec.begin(),vec.end());
vector<类型>::iterator it = unique(vec.begin(),vec.end());
vec.erase(it,vec.end());
4. C++程序
class Solution {
public:
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
vector<vector<int>> res;
vector<int> flag;
for(int i=0;i<nums.size();i++){
flag.push_back(0);
}
dfs(0,nums,flag,res);
sort(res.begin(),res.end());
vector<vector<int>>::iterator iter = unique(res.begin(),res.end());
res.erase(iter,res.end());
return res;
}
void dfs(int loc,vector<int> nums,vector<int> flag,vector<vector<int>>& res){
if(loc>=nums.size()){
vector<int> temp;
for(int i=0;i<nums.size();i++){
if(flag[i]==1) temp.push_back(nums[i]);
}
sort(temp.begin(),temp.end());
res.push_back(temp);
return;
}
for(int i=0;i<2;i++){
int old = flag[loc];
flag[loc] = i;
dfs(loc+1,nums,flag,res);
flag[loc] = old;
}
}
};