题目描述如下:
给定一组不含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
示例:
输入: nums = [1,2,3]
输出:
[
[3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[]
]
这是一道比较简单的题目,有很多种方法做,这里只介绍迭代法。
Python3代码如下:
class Solution:
def subsets(self, nums: List[int]) -> List[List[int]]:
res = [[]]
for i in reversed(nums):
res = res + [[i] + x for x in res]
return res
迭代法的代码很简单,但是有的朋友可能看不懂,这里我以nums=[1,2,3]为例给不明白的朋友演示一下。
初始,定义存储结果的列表并给列表赋一个值,即空集[]。
然后运行到下一行代码,这里开始循环。(reversed(nums)的意思是倒序遍历nums列表)
进入循环:
第一次循环,此时i=3,res=[[]](只有一个空集元素)
res = res + [[i] + x for x in res]
--> res = [[]] + [[3] + x for x in res]
--> res = [[]] + [[3] + []] = [[],[3]]
中括号里面的那个for循环只循环一次,因为res里只有一个空集元素
第二次循环,此时i=2,res=[[],[3]](此时res里有两个元素)
res = res + [[2] + x for x in res]
--> 1.res = [[],[3]] + [[2] + []]
--> 2.res = [[],[3]] + [[2] + [],[2] + [3]] = [[],[3],[2],[3,2]]
第三次循环,此时i=1,res=[[],[2],[3],[2,3]]
res = res + [[i] + x for x in res]
--> res = [[],[2],[3],[2,3]] + [[1] + []]
--> res = [[],[2],[3],[2,3]] + [[1] + [],[1] + [2]]
--> res = [[],[2],[3],[2,3]] + [[1] + [],[1] + [2],[1] + [3]]
--> res = [[],[2],[3],[2,3]] + [[1] + [],[1] + [2],[1] + [3],[1] + [2,3]]
= [[],[2],[3],[2,3]] + [[1],[1,2],[1,3],[1,2,3]]
= [[],[1],[2],[3],[1,2],[1,3],[2,3],[1,2,3]]
至此,[1,2,3]的所有子集已经得出