当前位置: 首页 > article >正文

[leetcode]113_路径总和II_输出所有路径

给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。
找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点

示例 1:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22
输出:[[5,4,11,2],[5,8,4,5]]

示例 2:
输入:root = [1,2,3], targetSum = 5
输出:[]
解释:树中存在两条根节点到叶子节点的路径:
(1 --> 2): 和为 3
(1 --> 3): 和为 4
不存在 sum = 5 的根节点到叶子节点的路径。
示例 3:
输入:root = [], targetSum = 0
输出:[]
解释:由于树是空的,所以不存在根节点到叶子节点的路径。

提示:
树中节点的数目在范围 [0, 5000] 内
-1000 <= Node.val <= 1000
-1000 <= targetSum <= 1000

解题思路:【DFS + 回溯】或者【BFS】

参考:

[leetcode]112_路径总和_判断是否存在-CSDN博客



 

import traceback
class TreeNode:
    def __init__(self, val, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

class Solution:
    def buildtree(self, levelorder = []):
        if not levelorder:
            return None
        root = TreeNode(levelorder[0])
        queue = [root]
        i = 1
        while queue and i < len(levelorder):
            node = queue.pop(0)
            if levelorder[i]:
                node.left = TreeNode(levelorder[i])
                queue.append(node.left)
            i += 1
            if i < len(levelorder) and levelorder[i]:
                node.right = TreeNode(levelorder[i])
                queue.append(node.right)
            i += 1
        return root
    def dfs(self, root, res, targetSum, tmpSum):
        if not root:
            return res
        if sum(tmpSum) == targetSum and not root.left and not root.right:
            res.append(tmpSum)
        if root.left:
            self.dfs(root.left, res, targetSum, tmpSum + [root.left.val])
        if root.right:
            self.dfs(root.right, res, targetSum, tmpSum + [root.right.val])
    def bfs(self, root, res, targetSum):
        if not root:
            return res
        queue = [(root, [root.val])]
        while queue:
            node, path = queue.pop(0)
            if not node.left and not node.right and sum(path) == targetSum:
                res.append(path)
            if node.left:
                queue.append((node.left, path + [node.left.val]))
            if node.right:
                queue.append((node.right, path + [node.right.val]))
if __name__ == '__main__':
    try:
        nums = eval(input())
        solution =Solution()
        root = solution.buildtree(nums)
        targetSum = int(input())
        res = []
        if root == None:
            print(res)
        else:
            # solution.dfs(root, res, targetSum, [root.val])
            solution.bfs(root, res, targetSum)
            print(res)
    except Exception as e:
        traceback.print_exc()

仅作为代码记录,方便自学自查自纠


http://www.kler.cn/a/318690.html

相关文章:

  • 应用于新能源汽车NCV4275CDT50RKG车规级LDO线性电压调节器芯片
  • WPF学习之路,控件的只读、是否可以、是否可见属性控制
  • MySql结合element-plus pagination的分页查询
  • 利用滑动窗口解题
  • 多叉树笔记
  • react动态路由
  • 【AIGC】ChatGPT RAG提取文档内容,高效制作PPT、论文
  • day-59 四数之和
  • 【React】(推荐项目)使用 React、Socket.io、Nodejs、Redux-Toolkit、MongoDB 构建聊天应用程序 (2024)
  • 数据集-目标检测系列-海洋鱼类检测数据集 fish>> DataBall
  • short-link笔记
  • ubuntu 24.04 输入设备显示没有,系统没有找到电脑麦克风
  • web平台搭建-LAMP(CentOS-7)
  • 【自动驾驶】基于车辆几何模型的横向控制算法 | Stanley 算法详解与编程实现
  • ai写论文哪个平台好?分享4款ai论文写作平台软件
  • Python范例总结
  • 【计算机视觉】YoloV8-训练与测试教程
  • javascript是什么语言?它是干什么的?
  • Element UI在工程中使用方式
  • 79、Python之鸭子类型:没有听过鸭子类型?关键在于认知的转变
  • 网络安全-长亭雷池waf的sql绕过,安全狗绕过(5种绕过3+2)
  • 安科瑞Acrel-1000DP分布式光伏监控系统在鄂尔多斯市鄂托克旗巴音乌苏六保煤矿5MW分布式光伏项目中的应用
  • [linux][证书]证书导出公钥
  • MySQL记录存储过程执行的错误信息
  • 改进拖放PDF转换为图片在转换为TXT文件的程序
  • 浅谈C++之多线程实现