给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
输入: [1,2,2]
输出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]
一、思路
(一)回溯算法
先对数组排序,再利用set的性质去重
C++代码:
class Solution {
public:
set<vector<int>> ans;
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
vector<int> subset;
sort(nums.begin(), nums.end());
trackingBack(nums, 0, subset);
return vector<vector<int>>(ans.begin(), ans.end());
}
void trackingBack(vector<int>& nums, int next_pos, vector<int>& subset) {
ans.insert(subset);
for (int i = next_pos; i < nums.size(); i++) {
subset.push_back(nums[i]);
trackingBack(nums, i + 1, subset);
subset.pop_back();
}
}
};
执行效率:
(二)不使用set如何去重?
其实也有不用set的方法,我们注意到重复时发生在pop之后,准确地说:
相同长度、相同位置的数字不能重复,因此在pop之后,应该对下一个数进行检查,看看是否相同
C++代码:
class Solution {
public:
vector<vector<int>> ans;
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
vector<int> subset;
sort(nums.begin(), nums.end());
for (int i = 0; i <= nums.size(); i++)
trackingBack(nums, 0, subset, i);
return ans;
}
void trackingBack(vector<int>& nums, int next_pos, vector<int>& subset, int subset_len) {
if (subset.size() == subset_len) {
ans.push_back(subset);
return;
}
for (int i = next_pos; i < nums.size(); i++) {
subset.push_back(nums[i]);
trackingBack(nums, i + 1, subset, subset_len);
subset.pop_back();
while(i + 1 < nums.size() && nums[i] == nums[i + 1])
i++;
}
}
};
执行效率: