给定一组不含重复元素的整数数组 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(n)种情况。因此 i<(1<<n) 其实就是 i<23.
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 二进制位,即枚举的所有情况实现。