LeetCode 90. 子集 II

最新推荐文章于 2024-06-02 14:57:21 发布

大白羊_Aries

最新推荐文章于 2024-06-02 14:57:21 发布

阅读量964

点赞数

分类专栏：题解文章标签： leetcode 算法职场和发展

本文链接：https://blog.youkuaiyun.com/qq_38204302/article/details/125155085

版权

题解专栏收录该内容

252 篇文章

订阅专栏

本文探讨了在处理含有重复元素的子集问题时，如何通过排序和剪枝优化回溯算法，避免冗余结果。通过实例说明并提供了C++代码实现，重点在于如何在`subsetsWithDup`函数中利用`backtrace`进行有效剪枝。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

90. 子集 II

解法

我们考虑回溯的方法，但是和 LeetCode 78. 子集不同的是我们这里需要考虑剪枝的问题，例如对于题目例子 $n u m s = [1, 2, 2]$ ，不剪枝的话生成的结果如下图所示
在这里插入图片描述
而正确的结果应该是

所以我们必须有剪枝这一步，体现在代码上，需要先进行排序，让相同的元素靠在一起，如果发现 $n u m s [i] = = n u m s [i - 1]$ ，则跳过。具体看下面实现

class Solution {
public:
    vector<vector<int>> res;

    vector<vector<int>> subsetsWithDup(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        vector<int> track;
        backtrace(nums, 0, track);
        return res;
    }

    void backtrace(vector<int>& nums, int start, vector<int> track)
    {
        res.push_back(track);

        for (int i = start; i < nums.size(); i++)
        {
            if (i > start && nums[i] == nums[i - 1]) continue;
            track.push_back(nums[i]);
            backtrace(nums, i + 1, track);
            track.pop_back();
        }
    }
};