78.子集 python

题目

题目描述

给你一个整数数组 nums ，数组中的元素 互不相同 。返回该数组所有可能的子集（幂集）。

解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。

示例 1：

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

示例 2：

输入：nums = [0]
输出：[[],[0]]

提示：

1 <= nums.length <= 10
-10 <= nums[i] <= 10
nums 中的所有元素 互不相同

题解

要生成一个数组的所有子集（幂集），可以使用回溯算法或迭代方法。这里，我将向你展示如何使用这两种方法来解决这个问题。

方法 1：回溯法

回溯法是一种通过构建所有可能的解决方案并逐步排除不符合条件的方案来解决问题的技术。对于子集问题，我们可以从空集开始，逐步添加元素，直到我们遍历了所有的元素。

def subsets(nums):
    def backtrack(start=0, current_subset=[]):
        # 添加当前子集到结果集中
        result.append(current_subset[:])
        for i in range(start, len(nums)):
            # 将当前元素添加到子集中
            current_subset.append(nums[i])
            # 使用递归继续构建下一个元素
            backtrack(i + 1, current_subset)
            # 回溯，移除最后添加的元素
            current_subset.pop()

    result = []
    backtrack()
    return result

提交结果

方法 2：迭代法

迭代法是通过逐步扩展已有的子集来生成新的子集。每次迭代时，我们将当前数字添加到现有的每个子集上，以创建新的子集。

def subsets(nums):
    result = [[]]  # 初始化结果集为包含空集
    for num in nums:
        # 对于每个数字，更新结果集，加入新生成的子集
        result += [current + [num] for current in result]
    return result

# 示例调用
print(subsets([1, 2, 3]))

提交结果

代码解释

• 回溯法：

  • 我们定义了一个辅助函数 backtrack，它接受两个参数：start 表示从哪个索引开始考虑元素，current_subset 是当前正在构建的子集。
  • 每次调用 backtrack 时，我们将当前子集复制一份并添加到结果列表中。
  • 然后，我们尝试将每一个后续的元素添加到当前子集中，并递归地对剩余的元素调用 backtrack。
  • 当递归返回时，我们执行回溯操作，即移除最后一个添加的元素，以便探索其他可能性。

• 迭代法：

  • 我们初始化结果集为 [[]]，即只有一个空集。
  • 对于输入数组中的每个元素，我们都遍历当前的结果集，并将该元素添加到现有子集上，从而创建新的子集。
  • 最终，我们将这些新生成的子集也添加到结果集中。

两种方法都能有效地生成所有子集，并且在最坏情况下的时间复杂度都是 O(2^n)，其中 n 是输入数组的长度，因为每个元素都有两种选择——要么在子集中，要么不在子集中。空间复杂度也是 O(2^n)，因为我们最终会存储所有的子集。