给你一个整数数组 nums
,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。
解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。
在解决这道题前,刚好复习一下切片的一些知识。
python切片:变量名[开始索引(默认为0):结束索引(默认为最后一个位置):步长(默认为1,为负时表示逆序)]
举例:
一维数组:[a:b],左闭右开
[::],所有元素
[::-1],倒序
二维数组:注意有无逗号
[:2],索引0-1行(前两行)
[:,2],所有行的第3列
[:,m:n] ,取二维数组的第m列到第n-1列所有行的数据
[m:n,:],取二维数组的第m行到第n-1行所有列的数据
三维数组:
a[0],首层
a[0][0],首层首行所有列
a[0][0][0],首层首行首列
a[:,0,0],所有层的首行首列
a[0,:,0],首层所有行首列
a[0,0,:],首层首行所有列
python列表切片在循环中要加[:],不然的话会反复循环,因为切片是引用新的对象,
此时在循环中res[:]是不更新的,而res是不断有元素push进去的。但有的情况可以不加,因为res一开始就固定住了,附上相关的情况代码链接:python列表切片加[:]情况
根据评论区大佬的思路:
可以根据这个思路,运用python切片:
class Solution:
def subsets(self, nums: List[int]) -> List[List[int]]:
result = [[]]
for i in range(len(nums)-1,-1,-1):
for j in result[:]:如果不进行切片[:],在后续append时会导致result更新,从而for会进入死循环
result.append(j+[nums[i]])
return result
尝试了正序遍历也是可以的,但是速度和空间性能会差一些,不知道为什么。
除此之外,刚刚提到的切片不加[:]的情况也可以应用,也就是迭代。
class Solution:
def subsets(self, nums: List[int]) -> List[List[int]]:
result = [[]]
for i in nums:
result += [[i] + k for k in result]
return result
官方有提到采用回溯法,那么什么是回溯法呢?
简单来说就是用树或图去模拟求解,所有的回溯算法,都可以看作一棵树从根节点向下遍历的过程,是一个纯暴力搜索法。可解决组合问题、切割问题、子集问题、排列问题、棋盘问题(如解数独)。回溯和递归相辅相成,一般在递归函数的下面。树的大小一般是集合大小(for循环),深度就是递归的深度。下面是处理的一个大致过程:
def 函数
if 终止条件:
收集结果
return
for循环(集合的元素集)
处理节点
递归函数
回溯(撤销处理节点的结果)
return
最后贴上大佬比较标准的回溯法代码:
class Solution:
def subsets(self, nums: List[int]) -> List[List[int]]:
res = []
path = []
def backtrack(nums,startIndex):
res.append(path[:]) #收集子集,要放在终止添加的上面,否则会漏掉自己
for i in range(startIndex,len(nums)): #当startIndex已经大于数组的长度了,就终止了,for循环本来也结束了,所以不需要终止条件
path.append(nums[i])
backtrack(nums,i+1) #递归
path.pop() #回溯
backtrack(nums,0)
return res
class Solution:
def subsets(self, nums: List[int]) -> List[List[int]]:
res = []
n = len(nums)
def helper(i, tmp):
res.append(tmp)
for j in range(i, n):
helper(j + 1,tmp + [nums[j]] )
helper(0, [])
return res
作者:powcai
链接:https://leetcode-cn.com/problems/subsets/solution/hui-su-suan-fa-by-powcai-5/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。