幂集

面试题 08.04. 幂集

幂集。编写一种方法，返回某集合的所有子集。集合中不包含重复的元素。

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

示例:

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

利用位运算:

nums = [1, 2, 3]，长度为3，取看成是1，不取看成是0.
那么就有：
[0, 0, 0] -> 0
[0, 0, 1] -> 1
[0, 1, 0] -> 2
[0, 1, 1] -> 3
[1, 0, 0] -> 4
[1, 0, 1] -> 5
[1, 1, 0] -> 6
[1, 1, 1] -> 7

把这些0和1看成是二进制，代表0~7共8个数字。幂集大小是2^nums.length。

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<vector<int>> res;
        size_t total = 1 << nums.size();
        for (size_t i = 0; i < total; ++i) {
            vector<int> vec;
            size_t num = i, idx = 0;
            while (num) {
                if (num & 1)此处需要注意:下图;
                    vec.push_back(nums[idx]);
                num >>= 1;
                ++idx;
            }
            res.push_back(vec);
        }
        return res;
    }
};

num&1:(与运算)最后一位是不是1;

num>>=1: 相当于: num=num>>1;

在循环中,若num=6(二进制:110):

6&1=0;num>>=1;(num=3)

3&1=1;num>>=1;(num=1)

1&1=1;num>>=1;(num=0)

与数组[a,b,c]正好对应其中;(110)

比较巧妙;

代码相同,但另一个比较巧妙的是:

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<vector<int>> res;此处是二维数组;
        size_t total = 1 << nums.size();
        for (size_t i = 0; i < total; ++i) {
            vector<int> vec;
            size_t num = i, idx = 0;
            while (num) {
                if (num & 1)
                    vec.push_back(nums[idx]);
                num >>= 1;
                ++idx;
            }
            res.push_back(vec);注意此处:相当于在一维之中的空间中加了一维;此处需要注意;
        }
        return res;
    }
};