来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subsets-ii
给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
输入: [1,2,2]
输出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]
题解:回溯法
class Solution {
public:
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
vector<vector<int>> results;
if (nums.size() == 0)
return results;
sort(nums.begin(), nums.end());
// 记录路径
vector<int> subset;
subsetHelper(nums, 0, subset, results);
return results;
}
private:
void subsetHelper(vector<int>& nums, int startIndex, vector<int>& subset, vector<vector<int>>& results) {
results.push_back(subset);
for (int i = startIndex; i < nums.size(); i ++) {
// 判断第i-1个元素与第i个元素是否相等且是否已经使用
if (i > 0 && nums[i] == nums[i - 1] && i != startIndex)
continue;
// 选择
subset.push_back(nums[i]);
// 回溯
subsetHelper(nums, i + 1, subset, results);
// 撤销选择
subset.pop_back();
}
}
};