LeetCode:78.子集

119 阅读1分钟

LeetCode:78.子集

题目描述

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

分析

简而言之,就是求集合的所有子集
可不就是初中题嘛,让我背一背公式,咳咳:2n2^n(包含∅)

原理是啥来着,每一个元素就两种状态嘛,选或者不选
看到了2,看到了两种状态,要素察觉:bit
没错,二进制

用n个比特,从0自增到2n12^n-1即可枚举所有子集状态

唯一的知识点只有:十进制转二进制
这个,不必我多说吧,移位运算符即可

Talk is Cheap. Show me the Code.

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        int N = nums.size();
        int sum = pow(2,N);
        vector<vector<int>> ans;
        for(int i = 0; i < sum; i++){
            int n = i;
            int k = 0;
            ans.push_back({});
            while(n){
                if(n&1)
                    ans.back().push_back(nums[k]);
                n>>=1;
                k++;
            }
        }
        return ans;
    }
};