力扣_数组28—子集

qweasdwxc

于 2024-01-11 20:57:10 发布

阅读量910

点赞数 24

文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/qweasdwxc/article/details/135538705

版权

本文介绍了一种使用回溯法解决编程问题的方法，目标是生成给定整数数组的所有可能子集，形成幂集。通过遍历每个元素并用二进制表示是否选择，生成不重复的子集列表。

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

题目

给你一个整数数组 nums ，数组中的元素互不相同。返回该数组所有可能的子集（幂集）。
解集不能包含重复的子集。你可以按任意顺序返回解集。

输入： $n u m s = [1, 2, 3]$
输出： $[[], [1], [2], [1, 2], [3], [1, 3], [2, 3], [1, 2, 3]]$

思路

回溯
$n u m s$ 中每一位选或者不选，构成了所有的子集，共 $2^n$ 种
- 每一位选或者不选可以用 $0, 1$ 表示
- 0~ $2^n-1$ 对应的二进制数对应了所有可能的子集
- 如假设数组长度为 $6$ ， $001101$ 表示选择第 $0, 2, 3$ 个数，其他不选

代码

class Solution {
public:
    // void subfind(vector<int>& nums, vector<int>& vec, vector<vector<int>>& ret, int max_len, int ind){
    //     if(vec.size() == max_len){
    //         ret.push_back(vec);
    //         return;
    //     }
    //     for(int i = ind; i < nums.size(); i++){
    //         vec.push_back(nums[i]);
    //         subfind(nums, vec, ret, max_len, i+1);
    //         vec.pop_back();
    //     }
    // }
    // vector<vector<int>> subsets(vector<int>& nums) {
    //     int n = nums.size();
    //     vector<int> vec;
    //     vector<vector<int>> ret;
    //     for(int i = 0; i <= n; i++){
    //         subfind(nums, vec, ret, i, 0);
    //     }
    //     return ret;
    // }

    vector<vector<int>> subsets(vector<int>& nums) {
        int n = nums.size();
        vector<vector<int>> ret;
        int mask_max = pow(2, n);
        for(int mask = 0; mask < mask_max; mask++){
            vector<int> vec;
            for(int i = 0; i < n; i++){
                if(mask & 1 << i){
                    vec.push_back(nums[i]);
                }
            }
            ret.push_back(vec);
        }
        return ret;
    }
};