LeetCode-Day99(C++) 78. 子集

最新推荐文章于 2024-08-25 22:13:13 发布

debu阿良

最新推荐文章于 2024-08-25 22:13:13 发布

阅读量78

点赞数

分类专栏： LeetCode手帐文章标签：递归回溯

本文链接：https://blog.csdn.net/weixin_45401129/article/details/119711274

版权

LeetCode手帐专栏收录该内容

130 篇文章 1 订阅

订阅专栏

这篇博客介绍了如何使用递归算法解决整数数组子集（幂集）的问题，给出了示例输入和输出，并详细解释了递归过程。在递归过程中，每个元素有两种状态（选中或不选中），通过记录已选中的元素和递归调用来生成所有可能的子集。时间复杂度为O(n×2^n)，空间复杂度为O(n)。

摘要由CSDN通过智能技术生成

子集
给你一个整数数组 nums ，数组中的元素互不相同。返回该数组所有可能的子集（幂集）。

解集不能包含重复的子集。你可以按任意顺序返回解集。

示例 1：

输入：nums = [1,2,3]
输出：[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]
示例 2：

输入：nums = [0]
输出：[[],[0]]

提示：

1 <= nums.length <= 10
-10 <= nums[i] <= 10
nums 中的所有元素互不相同

方法二：递归法实现子集枚举
思路与算法

我们也可以用递归来实现子集枚举。

假设我们需要找到一个长度为 nn 的序列 aa 的所有子序列，代码框架是这样的：

vector<int> t;
void dfs(int cur, int n) {
    if (cur == n) {
        // 记录答案
        // ...
        return;
    }
    // 考虑选择当前位置
    t.push_back(cur);
    dfs(cur + 1, n, k);
    t.pop_back();
    // 考虑不选择当前位置
    dfs(cur + 1, n, k);
}

上面的代码中，dfs(cur,n) 参数表示当前位置是 cur，原序列总长度为 n。原序列的每个位置在答案序列中的状态有被选中和不被选中两种，我们用 t 数组存放已经被选出的数字。在进入 dfs(cur,n) 之前[0,cur−1] 位置的状态是确定的，而 [cur,n−1] 内位置的状态是不确定的，dfs(cur,n) 需要确定 cur 位置的状态，然后求解子问题 dfs(cur+1,n)。对于 cur 位置，我们需要考虑 a[cur] 取或者不取，如果取，我们需要把 a[cur] 放入一个临时的答案数组中（即上面代码中的 t），再执行dfs(cur+1,n)，执行结束后需要对 t进行回溯；如果不取，则直接执行 dfs(cur+1,n)。在整个递归调用的过程中，cur 是从小到大递增的，当cur 增加到 n 的时候，记录答案并终止递归。可以看出二进制枚举的时间复杂度是 O(2 ^ n)

代码

class Solution {
public:
    vector<int> t;
    vector<vector<int>> ans;

    void dfs(int cur, vector<int>& nums) {
        if (cur == nums.size()) {
            ans.push_back(t);
            return;
        }
        t.push_back(nums[cur]);
        dfs(cur + 1, nums);
        t.pop_back();
        dfs(cur + 1, nums);
    }

    vector<vector<int>> subsets(vector<int>& nums) {
        dfs(0, nums);
        return ans;
    }
};