zl程序教程

您现在的位置是:首页 >  后端

当前栏目

122、【回溯算法】leetcode ——90. 子集 II:子集去重(C++版本)

C++LeetCode算法 版本 II 90 回溯 子集
2023-09-11 14:20:01 时间

题目描述

在这里插入图片描述
在这里插入图片描述
原题链接: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