【leetcode】子集(dfs,位运算)

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

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

示例:

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

链接：https://leetcode-cn.com/problems/subsets

dfs生成子集:

class Solution {
public:
    vector<vector<int>> ans;

    void dfs(int cur, vector<int>& a, vector<int>& nums){
        int n = nums.size();
            ans.push_back(a);
        for(int i = cur;i < n;i++){
            a.push_back(nums[i]);
            dfs(i+1,a,nums);
            a.pop_back();
        }
    }

    vector<vector<int>> subsets(vector<int>& nums) {
        int n = nums.size();
        if(!n) return ans;
        vector<int> a;
        dfs(0,a,nums);
        return ans;
    }
};

位运算压缩：

先上代码：

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

首先，这道题可以使用位运算的原理是：数组中每个位置的数都有“选与不选”两种状态，因此可以用二进制进行状态压缩。
对于下面的代码来说，i的含义其实就是枚举了所有的情况。
假设 n=3. 那么就有2⁽ⁿ⁾种情况。因此 i<(1<<n) 其实就是 i<2³.
i = 0 : 0 0 0
i = 1 : 0 0 1
i = 2 : 0 1 0
i = 3 : 0 1 1
i = 4 : 1 0 0
i = 5 : 1 0 1
i = 6 : 1 1 0
i = 7 : 1 1 1

对于第二层循环：

for(int k = 0, j = i;j > 0;j>>=1, k++){
                if(j&1 == 1){
                    tmp.push_back(nums[k]);
                }
            }

假设 i = 101，那么 j = 101 , k = 0.
（j&1 == 1) 表示判断 j 最后的二进制位是否为1，k则是用来匹配nums下标的。
第一轮：j = 101，末位为1，条件满足，nums[0] = 1 加进tmp，j 右移，k++;
第二轮：j = 10，末位为0，条件不满足，j 右移，k++;
第三轮：j = 1，末位为1，条件满足，nums[2] = 3 加进tmp，j 右移至0，k++;
循环结束，i = 101 对应的 tmp 数组为[1,3].

这样，就可以将所有的 i 二进制位，即枚举的所有情况实现。