算法设计与分析课程练习:
78. 子集
给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。
示例 1:
输入:nums = [1,2,3]
输出:[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]
示例 2:
输入:nums = [0]
输出:[[],[0]]
来源:力扣(LeetCode)
a) 算法思路
1、采用组合数学经常使用的思路,放或不放
2、比如1、2、3三个格子,从1开始,有两种选择,1选或者不选,然后再判断2,选或者不选。
3、两种选择分别进行递归
下面的回溯算法可以作为回溯的模板(感谢刘安老师提供的模板)
def subsets(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
n=len(nums)
res=[]
path=[]
def generate_subset(i):
if i>=n:
res.append(path[:])
else:
for c in range(2): #重点选或者不选
if c==0: #c==0表示选中
path.append(nums[i])
generate_subset(i+1) # 选要递归
path.pop()
else:
generate_subset(i+1) # 不选也要递归
generate_subset(0)
return res