LeetCode-78. 子集

最新推荐文章于 2024-07-22 16:55:07 发布

ガッシュ·ベル

最新推荐文章于 2024-07-22 16:55:07 发布

阅读量199

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/qq_41822647/article/details/87894617

版权

leetcode 专栏收录该内容

96 篇文章 0 订阅

订阅专栏

78. 子集

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

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

示例:

输入: nums = [1,2,3]
输出:
[
  [3],
  [1],
  [2],
  [1,2,3],
  [1,3],
  [2,3],
  [1,2],
  []
]

DFS

class Solution {
public:
    vector<vector<int>> vec;
    vector<int> v;
    vector<vector<int>> subsets(vector<int>& nums) {
        //vec.push_back(v);
        dfs(nums,0);
        return vec;
    }
    void dfs(vector<int>& nums,int k)
    {
        if(v.size() <= nums.size()) vec.push_back(v);
        else if(v.size()>nums.size()) return;
        for(int i = k;i<nums.size();i++)
        {
            v.push_back(nums[i]);
            //vec.push_back(v);
            dfs(nums,i+1);
            v.pop_back();
        }
    }
};

位运算

我们知道当集合中元素个数为n时，集合的子集个数为2的n次方，在位运算中，可以用0，1表示该位元素取还是不取。比如，集合{1,2,3},它的子集有

{1},{2},{3},{1,2},{1,3},{2,3},{1,2,3},{} //8个
//在位运算下对应的是
{001},{010},{011},{101},{110},{111},{000} //0表示不取该位上的元素，1表示取该位上的元素，011表示选取1,2构成一个子集

所以该问题可以转化为第i种情况，第j位上放相应的元素

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<vector<int>> vec;
        int sum = 1<<nums.size(); //2的n次方种
        for(int i=0;i<sum;i++)
        {
            vector<int> v;
            for(int j=0;j<nums.size();j++)
            {
                if(i & (1<<j)) //当i上的第j位为1时，表示该位上有代表的元素，那就在第j位上放nums[j]元素
                {
                    v.push_back(nums[j]);
                }
            }
            vec.push_back(v);
            v.clear();
        }
        return vec;
    }
};