来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subsets
给定一组不含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
输入: nums = [1,2,3]
输出:
[
[3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[]
]
分析:
含有n个元素的集合共有(1<<n)个子集,在选取每一个元素的时候,注意到每一个元素只有两种状态——选与不选,因此可以采用二进制表示这两种状态,具体来说就是用1表示该元素选,用0表示该元素不选。这样一来,每一种选法都能用唯一一个二进制数来表示。对于这个题目,只需要从0枚举到(1<<n)-1,对于其中的第i个数,对应唯一的二进制数。那么如何获得第i个数对应的选法呢?对于第i个数,采用先右移(>>)一位后与(&)1的方式判断最后一位是0还是1,如果是1,就把对应位置的数加入到vector中。
举个例子,给定集合[1,2,3,4,5,6,7,8,9],对于二进制数101100011,表示选择集合中的1,、2、6、7、9(注意是二进制数要倒着看)。
另外,注意一下本题中返回值的处理方式。由于题目要求返回二维vector,程序首先声明一个一维vector,然后声明一个二维vector,对于第i个数对应的状态,用一维vector保存,然后用二维vector保存一维vector,同时清空一维vector。
class Solution {
public:
vector<vector<int>> subsets(vector<int>& nums) {
int len=nums.size();
vector<int>ans1;
vector<vector<int>>ans2;
for(int i=0;i<=(1<<len)-1;i++){
int t=i;//十进制数i唯一对应二进制数
for(int j=0;j<=len-1;j++){
if(t&1){//当前位为1时
ans1.push_back(nums[j]);//内循环用一维vector记录第i个数对应的状态
}
t=t>>1;//右移一位,原来的倒数第二位变成了最后一位
}
ans2.push_back(ans1);//外循环用二维vector
ans1.clear();//内循环的一维vector清空
}
return ans2;
}
};