leetcode第78题,寻找给定集合的所有子集,包括空集。
一开始准备使用深搜,这也做倒也没啥问题,但是去重是一个很麻烦的事情。换种思路决定用二进制来做。
将给定集合的每个元素分别用一个二进制位代表,1表示该元素在子集合中,0表示不在子集中。这样所有二进制数的情况就对应了所有子集的可能情况,只要一一对应出来即可。
注意python的bin函数是转换二进制数,但是前面带着一个0b字符,所以我们要对它做一些处理,还要对齐数位,形成一个标准的字符串。
class Solution(object):
def subsets(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
n = len(nums)
binNum = 2**n
ans = []
for i in range(binNum):
s = "0"*(n-len(str(bin(i))[2:]))+str(bin(i))[2:]
print s
subAns = []
for j in range(n):
if s[j] == '1':
subAns.append(nums[j])
ans.append(subAns)
return ans
最近又学习了另外一种思路,回溯法,主要是用到了dfs,基本思路是,数组排序之后,每次按顺序向path中加入一个数字,加入后用dfs扩展后一位,每次进入递归函数时都要保存到最终结果数组中,当扩展到最后一位时,函数返回到上一层,上一层函数把最后一位删去,再引进下一位。
class Solution:
def subsets(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
n = len(nums)
nums.sort()
def dfs(nums,pos, path, ans):
import copy
ans.append(copy.deepcopy(path))
for i in range(pos,len(nums)):
path.append(nums[i])
dfs(nums, i+1, path, ans)
path.pop()
ans = []
dfs(nums, 0, [], ans)
return ans