leetcode 124. 二叉树中的最大路径和

二叉树中的 路径 被定义为一条节点序列,序列中每对相邻节点之间都存在一条边。同一个节点在一条路径序列中 至多出现一次 。该路径 至少包含一个 节点,且不一定经过根节点。

路径和 是路径中各节点值的总和。

给你一个二叉树的根节点 root ,返回其 最大路径和 。

示例 1:

输入:root = [1,2,3]

输出:6

解释:最优路径是 2 -> 1 -> 3 ,路径和为 2 + 1 + 3 = 6

示例 2:

输入:root = [-10,9,20,null,null,15,7]

输出:42

解释:最优路径是 15 -> 20 -> 7 ,路径和为 15 + 20 + 7 = 42

提示:

树中节点数目范围是 [1, 3 * 104]

-1000 <= Node.val <= 1000

题目链接在:leetcode 124

思路,可以通过一个函数,求以 root 为 根节点到 子节点的最大 pathSum, 这样递归遍历就能求出所有的 pathSum

python 复制代码
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def __init__(self):
        self.res = -pow(2, 31)
        self.a = dict() ## 存以 root 为根节点,一直到子节点的最大 pathSum

    ## 计算以 root 节点为根节点 往某一边的最大 pathSum
    @cache
    def getPathSum(self, root):
        if root == None:
            return 0
        #leftPathSum = self.a[root.left] if root.left in self.a else max(self.getPathSum(root.left), 0)
        #rightPathSum = self.a[root.right] if root.right in self.a else max(self.getPathSum(root.right), 0)
        leftPathSum =  max(self.getPathSum(root.left), 0)
        rightPathSum = max(self.getPathSum(root.right), 0)
        self.res = max(self.res, root.val + leftPathSum + rightPathSum)
        #self.a[root] = root.val + max(leftPathSum, rightPathSum)
        return root.val + max(leftPathSum, rightPathSum)


    def maxPathSum(self, root: Optional[TreeNode]) -> int:
        x = self.getPathSum(root)
        return self.res
相关推荐
VertexGeek24 分钟前
Rust学习(八):异常处理和宏编程:
学习·算法·rust
石小石Orz25 分钟前
Three.js + AI:AI 算法生成 3D 萤火虫飞舞效果~
javascript·人工智能·算法
jiao_mrswang1 小时前
leetcode-18-四数之和
算法·leetcode·职场和发展
qystca1 小时前
洛谷 B3637 最长上升子序列 C语言 记忆化搜索->‘正序‘dp
c语言·开发语言·算法
薯条不要番茄酱1 小时前
数据结构-8.Java. 七大排序算法(中篇)
java·开发语言·数据结构·后端·算法·排序算法·intellij-idea
今天吃饺子2 小时前
2024年SCI一区最新改进优化算法——四参数自适应生长优化器,MATLAB代码免费获取...
开发语言·算法·matlab
是阿建吖!2 小时前
【优选算法】二分查找
c++·算法
王燕龙(大卫)2 小时前
leetcode 数组中第k个最大元素
算法·leetcode
不去幼儿园3 小时前
【MARL】深入理解多智能体近端策略优化(MAPPO)算法与调参
人工智能·python·算法·机器学习·强化学习
Mr_Xuhhh3 小时前
重生之我在学环境变量
linux·运维·服务器·前端·chrome·算法