leetcode 腾讯精选50题 子集

题目解析

题目如下：
给定一组不含重复元素的整数数组 nums，返回该数组所有可能的子集（幂集）。
说明：解集不能包含重复的子集。
示例:
输入: nums = [1,2,3]
输出:
[ [3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[] ]

思路：
位运算法。
一个大小为n且没有重复元素的数组，他的子集一共有2^n个，而且每一个整数对应的二进制数都可以表示为一个子集（其中1代表取这个位置的数，0代表不取）。如果j对应为1，那么就说明当前解有第j位元素。

代码实现

class Solution {
public:
    vector<vector<int>> subsets(vector<int>& nums) {
        vector<vector<int>> ans;
        int n = nums.size();
        for(int i = 0; i < 1 << n; i++)
        {
            vector<int> curr;
            for(int j = 0; j < n; j++)
            {
                if(i >> j & 1 == 1)
                    curr.push_back(nums[j]);
            }
            ans.push_back(curr);
        }    
        return ans;
    }
};