题目链接:https://leetcode-cn.com/problems/subsets/
description:
给定一组不含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
输入: nums = [1,2,3]
输出:
[
[3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[]
]
方法:
- 思路1: 非递归
- 思路2: 回溯:基本形式 “递归”+“循环”
思路1:
我们可以一位一位的往上叠加,比如对于题目中给的例子[1,2,3]来说,最开始是空集,那么我们现在要处理1,就在空集上加1,为[1],现在我们有两个子集[]和[1],下面我们来处理2,我们在之前的子集基础上,每个都加个2,可以分别得到[2],[1, 2],那么现在所有的子集合为[], [1], [2], [1, 2],同理处理3的情况可得[3], [1, 3], [2, 3], [1, 2, 3], 再加上之前的子集就是所有的子集合了。
python3代码实现:
import copy
class Solution(object):
def subsets(self, nums):
"""
:type nums: List[int]
:rtype: List[List[int]]
"""
res = [[]]
global pre
for i, v in enumerate(nums):
size = len(res)
pre = copy.deepcopy(res)
for j in range(size):
res[j].append(v)
res = res + pre
return res
nums = [1, 2, 3]
s = Solution()
result = s.subsets(nums)
print(result[-8:])
上面代码简化后:
class Solution:
def subsets(self, nums):
ans = [[]]
for e in nums:
ans = ans + [(x + [e]) for x in ans]
return ans
nums = [1, 2, 3]
s = Solution()
result = s.subsets(nums)
print(result)
思路2:
本解法采用回溯算法实现,回溯算法的基本形式是“递归+循环”,正因为循环中嵌套着递归,递归中包含循环,这才使得回溯比一般的递归和单纯的循环更难理解,其实我们熟悉了它的基本形式,就会觉得这样的算法难度也不是很大。
python3代码:
class Solution(object):
def subsets(self, nums):
res = []
def dfs(j, tmp):
res.append(tmp)
for i in range(j, len(nums)):
dfs(i + 1, tmp + [nums[i]])
dfs(0, [])
return res
思路3:(参考)
由于原集合每一个数字只有两种状态,要么存在,要么不存在,那么在构造子集时就有选择和不选择两种情况,所以可以构造一棵二叉树,左子树表示选择该层处理的节点,右子树表示不选择,最终的叶节点就是所有子集合。
[] / \ / \ / \ [1] [] / \ / \ / \ / \ [1 2] [1] [2] [] / \ / \ / \ / \ [1 2 3] [1 2] [1 3] [1] [2 3] [2] [3] []