思路
题目要求找子集,且给定数据无重复元素。由数学知识有子集个数为
2
n
2^n
2n个,一个想法是用一个二进制数表示一种子集的状态
即可!!!
什么是状态?
如给定集合s = { 1, 2, 3 }
,(不考虑集合的无序性,将集合当做有序的)
- 选取
零个
元素,用二进制数000
表示 - 选取一个元素
2
,用二进制数010
表示 - 选取两个元素
1
和3
,用二进制数101
表示 全选
元素,用二进制数111
表示
需要做什么?
选取元素?No!!! 因为选取元素是我们最后的结果!!!
正确的打开方式是通过二进制数表示(即状态)作为纽带,得到集合中该选取的下标。
如给定二进制表示101
,判断哪个元素不为0
我们就将这个元素对应的下标
从集合s中拿出来,还是上面的例子,我们需要拿到的子集为[1, 3]。
找到一个二进制数中的1
例子:
二进制数表示:100110101001
如果能依次用:000000000001
如果能依次用:000000000010
.......
如果能依次用:100000000000
和上面的二进制数表示相与就知道哪里为1
观察数据的变化知道是将1每次左移动一位,然后和二进制数表示相与
,需要移动的次数为二进制数的长度。
code
class Solution:
def subsets(self, nums) :
# 使用位运算
size = len(nums)
maxi = 1 << size
ret = []
for i in range(maxi):
temp = []
for j in range(size):
if (1 << j) & i != 0:
temp.append(nums[j]) # 找到不为0的下标并将其从nums中取出
ret.append(temp)
return ret