78 子集 - 技术栈

78 子集

Rocoberry_团子2023-10-30 18:22

子集

- [题解1 回溯](#题解1 回溯)
- - 模板
- [题解2 迭代（mask思想）](#题解2 迭代（mask思想）)

给你一个整数数组 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 中的所有元素互不相同

题解1 回溯

cpp 复制代码

class Solution {
    vector<vector<int>> ret;
    deque<int> tmp;
public:
    void backtrace(vector<int>& nums, int len, int idx){
        if(len > nums.size()) return;
        if(tmp.size() == len){
            ret.push_back(vector<int>(tmp.begin(), tmp.end()));
            // idx 此时还没用过
            backtrace(nums, len+1, idx);
            return;
        }
        for(int i = idx; i < nums.size(); i++){
            tmp.push_back(nums[i]);
            backtrace(nums, len, i+1);
            tmp.pop_back();
        }
    }
    vector<vector<int>> subsets(vector<int>& nums) {
        backtrace(nums, 0, 0);
        return ret;
    }
};

模板

cpp 复制代码

class Solution {
    vector<vector<int>> ret;
    vector<int> tmp;
public:
    void backtrace(vector<int>& nums, int idx){
        if(idx == nums.size()){
            ret.push_back(tmp);
            return;
        }
        tmp.push_back(nums[idx]);
        // 选
        backtrace(nums, idx+1);
        tmp.pop_back();
        // 不选
        backtrace(nums, idx+1);

    }
    vector<vector<int>> subsets(vector<int>& nums) {
        backtrace(nums, 0);
        return ret;
    }
};

题解2 迭代（mask思想）

cpp 复制代码

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<vector<int>> ret;
        vector<int> tmp;
        int s = nums.size();
        // 每个子集对应一个长度为s的01序列，第i位表示nums[i]是否在子集中
        // mask 的值代表tmp当前有多少个元素，以及nums哪几个元素在tmp里
        for(int mask = 0; mask < 1 << s; mask ++){
            tmp.clear();

            for(int i = 0; i < s; i++)
            // 根据mask的值把元素放到tmp里
                if(mask & (1<<i)) 
                    tmp.push_back(nums[i]);
            
            ret.push_back(tmp);
        }
        return ret;
    }
};