练习三
子集
给定一组不含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
输入: nums = [1,2,3]
输出:
[
[3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[]
]
两种思路
- 位算法:[1,2,3]
二进制 | 对应子集 |
---|---|
000 | [] |
001 | [3] |
010 | [2] |
011 | [2,3] |
100 | [1] |
101 | [1,3] |
110 | [1,2] |
111 | [1,2,3] |
def func(nums):
xl = []
for i in range(2**len(nums)):
#与二进制格式相同,转为二进制
nl = []
n = bin(i)[2:]
# 右补齐
ns = n.rjust(len(bin(2**len(nums)-1)[2:]), '0')
nl.extend(ns)
#'1'出现的坐标,添加到yl
location = 0
yl = []
for i in range(nl.count('1')):
location += nl[location:].index('1')
yl.append(nums[location])
location += 1
xl.append(yl)
return xl
- 回溯
回溯是递归的一种,就不在这里板门农夫讲解递归了。一句话说来就是套娃,打开在套回去。
回溯算法是数据结构中树中的一种算法,基础的东西还得努力去看,尽量把别的语言实现的转为自己要用的语言写一遍
def subsets(self, nums: List[int]) -> List[List[int]]:
ans = []
# 存储符合要求的子集
tmp = []
n = len(nums)
def helper(idx):
# 先添加子集
ans.append(tmp[:])
for i in range(idx, n):
tmp.append(nums[i])
# 避免重复,每次递归,从下一个索引开始
helper(i+1)
# 回溯
tmp.pop()
helper(0,0)
return ans