回溯 | 13 90.子集II（树层去重&树枝去重）

本文记录的是刷题过程中的重要概念和笔记。如有侵权，请联系删除。

90.子集II

力扣题目链接(opens new window)

给定一个可能包含重复元素的整数数组 nums，返回该数组所有可能的子集（幂集）。

说明：解集不能包含重复的子集。

示例:

输入: [1,2,2]
输出: [ [2], [1], [1,2,2], [2,2], [1,2], [] ]

思路

树层去重

class Solution
{
public:
    vector<vector<int>> res;
    vector<int> path;
    void backtracing(vector<int> nums, int startIndex)
    {
        res.push_back(path); // 一开始就push，就不需要再额外加入空集了
        // 可以不需要加终止条件，因为startIndex >= nums.size()，本层for循环本来也结束了
        // if(startIndex==nums.size()){
        //     return;
        // }
        for (int i = startIndex; i < nums.size(); i++)
        {
            if (i - 1 >= startIndex && nums[i] == nums[i - 1])
                continue;
            path.push_back(nums[i]);
            backtracing(nums, i + 1);
            path.pop_back();
        }
        // 求取子集问题，不需要剪枝
    }
    vector<vector<int>> subsetsWithDup(vector<int> &nums)
    {
        res.clear();
        path.clear();
        sort(nums.begin(), nums.end());
        backtracing(nums, 0);
        return res;
    }
};

used法

关键：

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();
        }

set法

关键

void backtracking(vector<int>& nums, int startIndex) {
        result.push_back(path);
        unordered_set<int> uset;
        for (int i = startIndex; i < nums.size(); i++) {
            if (uset.find(nums[i]) != uset.end()) {
                continue;
            }
            uset.insert(nums[i]);
            path.push_back(nums[i]);
            backtracking(nums, i + 1);
            path.pop_back();
        }
    }

使用set针对同一父节点本层去重，但子集问题一定要排序，为什么呢？

我用没有排序的集合{2,1,2,2}来举例子画一个图，如图：

分析

使用set去重，不仅时间复杂度高了，空间复杂度也高了
组合，子集，排列问题的空间复杂度都是O(n)，但如果使用set去重，空间复杂度就变成了O(n^2)，因为每一层递归都有一个set集合，系统栈空间是n，每一个空间都有set集合。

用used数组也是占用O(n)的空间
used数组可是全局变量，每层与每层之间公用一个used数组，所以空间复杂度是O(n + n)，最终空间复杂度还是O(n)。