给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
输入: [1,2,2]
输出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]
? https://leetcode-cn.com/problems/subsets-ii
对于重复n次的数字可以出现0-n次,依次枚举就好了。
class Solution {
public:
vector<vector<int>>res;
vector<int> pro;
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
sort(nums.begin(),nums.end());
dfs(nums,0);
return res;
}
void dfs(vector<int>& nums,int u){
if (u == nums.size()){
res.push_back(pro);
return;
}
// 找出该元素重复出现了多少次
int k = 0;
while(u + k < nums.size() && nums[u+k] == nums[u]) ++k;
for(int i = 0 ; i <= k ; ++i){ // 从出现0次 到 n次
dfs(nums,u+k);
pro.push_back(nums[u]);
}
for (int i = 0 ; i <= k ; ++i) // 回溯
pro.pop_back();
}
};