leetcode 子集 python3

给定一组不含重复元素的整数数组 nums，返回该数组所有可能的子集（幂集）。

说明：解集不能包含重复的子集。

示例：
输入: nums = [1,2,3]
输出:
[
  [3],
  [1],
  [2],
  [1,2,3],
  [1,3],
  [2,3],
  [1,2],
  []
]

迭代
class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        res = [[]]
        for i in nums:
            res += [[i] + num for num in res]
        return res

其实回溯算法关键在于:不合适就退回上一步

然后通过约束条件, 减少时间复杂度.

下面转自与：https://leetcode-cn.com/problems/subsets/solution/hot-100-78zi-ji-python3-hui-su-liang-chong-jie-ti-/

思路一：

回溯三要素

1.有效结果

当指向元素的index==len(nums)的时候，就说明这一次的搜索结束了

if index == len(nums):
    self.res.append(sol)
    return

2.回溯范围及答案更新

不需要循环遍历，而只需要用一个index指向每一次的元素，下一层更新index = index+1
对于答案更新，我们需要考虑选或不选当前答案，保存当前index指向的元素

self.backtrack(sol+[nums[index]], index+1, nums)  
self.backtrack(sol, index+1, nums)

3.剪枝条件

不需要剪枝

代码

class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        self.res = []
        self.backtrack([], 0, nums)
        return self.res
        
    def backtrack(self, sol, index, nums):
        if index == len(nums):
            self.res.append(sol)
            return
        
        self.backtrack(sol+[nums[index]], index+1, nums)
        self.backtrack(sol, index+1, nums)

思路二：

class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        def backtrack(first = 0, curr = []):
            # if the combination is done
            if len(curr) == k:  
                output.append(curr[:])
            for i in range(first, n):
                # add nums[i] into the current combination
                curr.append(nums[i])
                # use next integers to complete the combination
                backtrack(i + 1, curr)
                # backtrack
                curr.pop()
        
        output = []
        n = len(nums)
        for k in range(n + 1):
            backtrack()
        return output

算法

定义一个回溯方法 backtrack(first, curr)，第一个参数为索引 first，第二个参数为当前子集 curr。

如果当前子集构造完成，将它添加到输出集合中。

否则，从 first 到 n 遍历索引 i。

将整数 nums[i] 添加到当前子集 curr。

继续向子集中添加整数：backtrack(i + 1, curr)。

从 curr 中删除 nums[i] 进行回溯。

https://leetcode-cn.com/problems/subsets/solution/zi-ji-by-leetcode/