力扣题目:给定一组可能含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例: 输入: nums = [1,2,2] 输出: [ [], [1], [2] , [1,2], [2,2], [1,2,2] ]
思路:每次都是从剩余集合中取元素加到已有组合好的子集里,如图解所示,考虑用递归函数,
递归函数:(1)确定参数(2)确定终止条件(3)确定单层处理逻辑
(1)确定参数
由于每次都是从剩余集合中取元素加到已有组合好的子集里,所以单层处理逻辑里面要用到剩余集合里面元素的下标,所以有个参数startIndex,跟用vec容器装着的集合nums,startIndex是在nums中的下标。
(2)确定终止条件
当剩余集合为空时,结束递归。即:当传进来的参数startIndex超出nums的下标表示范围时,结束递归。
(3)确定单层处理逻辑
在剩余集合里面取元素取元素,加进path容器里存储,取完剩余集合里面所有元素。每一次从剩余集合里面取元素取一个元素,把元素加进path,由于集合里可能包含重复元素,需要用used容器标记,如果要选取的元素与上一个元素相同(需要先对集合排序),并且上一个元素在used里标记为0,说明上一个元素未加进到该path里,说明是重复元素出现在同一个树层上,不可以选取该元素。可选取该元素后,再调用递归函数,进入下一个元素的选取,由于取过的元素不能重复选取加进到path里,所以需要回溯,把该元素从path中取出来。
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex, vector<bool>& used) {
result.push_back(path);
for (int i = startIndex; i < nums.size(); i++) {
// used[i - 1] == true,说明同一树枝candidates[i - 1]使用过
// used[i - 1] == false,说明同一树层candidates[i - 1]使用过
// 而我们要对同一树层使用过的元素进行跳过
if (i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) {
continue;
}
path.push_back(nums[i]);
used[i] = true;
backtracking(nums, i + 1, used);
used[i] = false;
path.pop_back();
}
}
public:
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
result.clear();
path.clear();
vector<bool> used(nums.size(), false);
sort(nums.begin(), nums.end()); // 去重需要排序
backtracking(nums, 0, used);
return result;
}
};