回溯——6.子集

力扣题目链接

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

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

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

解题思路总结

  • 使用回溯法(backtracking)来解决子集问题,其核心思路是对每一个元素做两种选择:要么将其加入到当前的子集中,要么不加入,并不断递归处理剩下的元素。
  • 在每一次递归调用时,都会将当前构造的 path 加入到 result 中,确保所有可能的子集都被枚举出来。
  • 最终返回的 result 包含了所有的子集,包括空集。

完整代码如下:

python 复制代码
class Solution:
    def subsets(self, nums):
        result = []
        path = []
        self.backtracking(nums, 0, path, result)
        return result

    def backtracking(self, nums, startIndex, path, result):
        result.append(path[:])  # 收集子集,要放在终止添加的上面,否则会漏掉自己
        # if startIndex >= len(nums):  # 终止条件可以不加
        #     return
        for i in range(startIndex, len(nums)):
            path.append(nums[i])
            self.backtracking(nums, i + 1, path, result)
            path.pop()
python 复制代码
class Solution:
    def subsets(self, nums):
        result = []
        path = []
        self.backtracking(nums, 0, path, result)
        return result
  • nums:输入的数组,表示需要找出其所有子集的集合。
  • result:用于存储最终结果,即所有可能的子集。
  • path:用来存储当前递归路径上选择的元素,代表一个子集。

在这个函数中,首先定义了 resultpath,分别用来存储最终所有子集的集合和当前正在构建的子集,然后调用 backtracking 函数开始回溯。backtracking 函数会从 nums 的第一个元素(下标为 0)开始递归地探索所有子集。

python 复制代码
    def backtracking(self, nums, startIndex, path, result):
        result.append(path[:])  # 收集子集
        for i in range(startIndex, len(nums)):
            path.append(nums[i])  # 选择元素加入子集
            self.backtracking(nums, i + 1, path, result)  # 递归调用
            path.pop()  # 撤销选择,回溯
2.1 result.append(path[:])
  • 这行代码表示:将当前路径 path 中的元素加入到 result 中。注意这里的 path[:] 是一个浅拷贝,确保后续对 path 的修改不会影响已经加入 result 的内容。
  • 由于任何一个时刻的 path 都可以作为一个合法的子集,因此每当进入 backtracking 函数时,无论 path 中有多少元素,都会将它作为一个子集存入 result
2.2 for 循环
  • for i in range(startIndex, len(nums)):表示从 nums 中当前的起始位置 startIndex 开始,依次选择元素加入到当前子集(即 path)中。
  • 对于每一个元素 nums[i],首先将其加入到 path 中,然后递归调用 backtracking 函数,开始探索从 i + 1 位置开始的下一个元素。
  • 递归完毕后,通过 path.pop() 将最后加入的元素删除,进行回溯,尝试其他的可能组合。
2.3 终止条件
  • 代码中没有显式的终止条件。事实上,回溯函数的退出是通过遍历 nums 数组的长度自动完成的。当 startIndex 达到 len(nums) 时,for 循环不再执行,因此递归自然结束。
  • 可以加一个显式的终止条件,但实际上没有这个必要,因为 for 循环本身已经会停止。
相关推荐
写个博客14 分钟前
代码随想录算法训练营第三十九天
算法
源码方舟1 小时前
【基于ALS模型的教育视频推荐系统(Java实现)】
java·python·算法·音视频
fancy1661662 小时前
力扣top100 矩阵置零
人工智能·算法·矩阵
元亓亓亓2 小时前
LeetCode热题100--240.搜索二维矩阵--中等
算法·leetcode·矩阵
明月看潮生3 小时前
青少年编程与数学 02-019 Rust 编程基础 09课题、流程控制
开发语言·算法·青少年编程·rust·编程与数学
oioihoii3 小时前
C++23 views::slide (P2442R1) 深入解析
linux·算法·c++23
yuhao__z4 小时前
代码随想录算法训练营第六十三天| 图论9—卡码网47. 参加科学大会,94. 城市间货物运输 I
算法·图论
June`4 小时前
专题三:穷举vs暴搜vs深搜vs回溯vs剪枝(全排列)决策树与递归实现详解
c++·算法·深度优先·剪枝
vlln4 小时前
适应性神经树:当深度学习遇上决策树的“生长法则”
人工智能·深度学习·算法·决策树·机器学习
冲帕Chompa5 小时前
图论part09dijkstra算法
算法·图论