给你一个整数数组 nums ,其中可能包含重复元素,请你返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。返回的解集中,子集可以按 任意顺序 排列。
示例 1:
输入:nums = [1,2,2]
输出:[[],[1],[1,2],[1,2,2],[2],[2,2]]
示例 2:
输入:nums = [0]
输出:[[],[0]]
我一开始做这道题的思路是,先对nums排序,再直接递归找到所有的组合数,然后用set去重。但显然这不是一个好方法,因为会搜索到无用的序列。
另一种想法是当回溯做弹出后,如果当前元素和下一个元素相等,就跳过深层递归,对pos 加1。比如 1 2 2 3。当下标为1的元素弹出后,因为nums[1]和nums[2]相等,所以跳过下标为2的深层递归,对下标为3做深层递归。这样做的道理是:去掉一个2的序列在第二个元素被包含时已经有了。第二个元素被弹出,第3个元素不应该再被包含,这样就直接跳过了第三个元素的递归。
class Solution {
public:
vector<vector<int>> ret;
vector<int> nums;
vector<int> temp;
void Recursion(int pos)
{
if (pos == nums.size()) {
ret.push_back(temp);
return;
}
temp.push_back(nums[pos]);
Recursion(pos + 1);
temp.pop_back();
while (pos + 1 < nums.size() && nums[pos + 1] == nums[pos]) {
++pos;
}
Recursion(pos + 1);
}
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
sort(nums.begin(), nums.end());
this -> nums = nums;
Recursion(0);
return ret;
}
};