题目描述
原题链接:90. 子集 II
解题思路
本题相对于 78. 子集(子集问题) 的区别在于,数组中可能会含有相同元素,可允许单个结果内有重复,但若还是获取所有树结点的话,会出现结果与结果之间有重复。因此,需要对会出现重复的结果进行去重。
去重方式与 40.组合总和II(bool型变量方法+startIndex去重方法) 相同,可采用判定stratIndex
是否大于i
与判定nums[i]
与nums[i - 1]
是否相同。若大于且相同,则说明有重复元素,因在没有遍历到这个元素之前,已经将这一数字与其余数字的组合结果存储过,因此对于出现相同数字时,跳
过该种情况即可。
1、startIndex去重
class Solution {
public:
vector<int> path;
vector<vector<int>> res;
void backtracking(vector<int>& nums, int startIndex) {
res.push_back(path);
if(startIndex == nums.size()) return ;
for(int i = startIndex; i < nums.size(); i++) {
// 保存第一次出现某一数字的情况,跳过出现重复数字的情况
if(i > startIndex && nums[i] == nums[i - 1]) continue;
path.push_back(nums[i]);
backtracking(nums, i + 1);
path.pop_back();
}
}
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
sort(nums.begin(), nums.end()); // 去重需要排序,保证相同元素相邻排列
backtracking(nums, {});
return res;
}
};
2、visited去重
class Solution {
public:
vector<int> path;
vector<vector<int>> res;
void backtracking(vector<int>& nums, int startIndex, vector<bool> visited) {
res.push_back(path);
if(startIndex == nums.size()) return ;
for(int i = startIndex; i < nums.size(); i++) {
// i > 0:保证不越界,nums[i] == nums[i - 1]:判定是否有重复数字,visited[i - 1] == false:说明第一个出现的数字已经被遍历过,此时为重复元素开始遍历,需去重
if(i > 0 && nums[i] == nums[i - 1] && visited[i - 1] == false) continue;
visited[i] = true;
path.push_back(nums[i]);
backtracking(nums, i + 1, visited);
path.pop_back();
visited[i] = false;
}
}
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
vector<bool> visited(nums.size(), false);
sort(nums.begin(), nums.end()); // 去重需要排序,保证相同元素相邻排列
backtracking(nums, {}, visited);
return res;
}
};
参考文章:90.子集II