LeetCode(python)78.子集

题目

给你一个整数数组 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.递归函数要做什么

(1)对每个数字进行决策,选还是不选

(2)把决策后的子集收集到最终的结果中

2.终止条件------当所有数字都决策过了,就存入此次递归的结果

3.单层逻辑

处理当前第index个元素时,有两种选择:

(1)选当前元素->加入临时子集->继续决策下一个元素

(2)不选当前元素->把它从临时子集中剔除->继续决策下一个元素(回溯)

代码

注意!res.append(temp[:]),不能写成res.append(temp),这样存的是temp的引用,后续会随着递归而改变

python 复制代码
class Solution:
    def subsets(self, nums: List[int]) -> List[List[int]]:
        res = []    # 存结果
        temp = []   # 存临时子集

        def dfs(index):
            if index == len(nums):        # 终止条件
                res.append(temp[:])       # 注意!一定要加:,不然存的就是temp的引用,会随着递归时temp的变换而变换,导致最后全为空
                return
            temp.append(nums[index])      # 选当前元素
            dfs(index + 1)                # 继续决策下一个

            temp.pop()                    # 不选
            dfs(index + 1)                # 继续决策下一个
        
        dfs(0)                            # 从第一个元素开始处理
        return res
相关推荐
地平线开发者7 小时前
SparseDrive 模型导出与性能优化实战
算法·自动驾驶
董董灿是个攻城狮7 小时前
大模型连载2:初步认识 tokenizer 的过程
算法
地平线开发者8 小时前
地平线 VP 接口工程实践(一):hbVPRoiResize 接口功能、使用约束与典型问题总结
算法·自动驾驶
罗西的思考8 小时前
AI Agent框架探秘:拆解 OpenHands(10)--- Runtime
人工智能·算法·机器学习
孟健9 小时前
Karpathy 用 200 行纯 Python 从零实现 GPT:代码逐行解析
python
HXhlx11 小时前
CART决策树基本原理
算法·机器学习
码路飞11 小时前
写了个 AI 聊天页面,被 5 种流式格式折腾了一整天 😭
javascript·python
Wect11 小时前
LeetCode 210. 课程表 II 题解:Kahn算法+DFS 双解法精讲
前端·算法·typescript
颜酱12 小时前
单调队列:滑动窗口极值问题的最优解(通用模板版)
javascript·后端·算法
曲幽14 小时前
FastAPI压力测试实战:Locust模拟真实用户并发及优化建议
python·fastapi·web·locust·asyncio·test·uvicorn·workers