124. 二叉树中的最大路径和(困难)-二叉树

二叉树中的 路径 被定义为一条节点序列,序列中每对相邻节点之间都存在一条边。同一个节点在一条路径序列中 至多出现一次 。该路径 至少包含一个 节点,且不一定经过根节点。
路径和 是路径中各节点值的总和。给你一个二叉树的根节点 root ,返回其 最大路径和 。

分析

  • 变化的变量太多,需要固定一些,减少重复就可以进一步减少运算时间
  • 可以定义一个最小单位,经过当前节点,并且是以当前节点为起点或者终点的路径
  • 这样的的话,就可以分成四种情况,就是要嘛自己最大,要嘛加上左节点的最大路径最大,要嘛加上有节点的最大路径最大,要嘛左右节点的最大
  • 但是返回只能返回前三种,有个全局变量记录四种的最大
# 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 maxPathSum(self, root: Optional[TreeNode]) -> int:

        self.max_sum = float("-inf")
        self.dfs(root)

        return self.max_sum


        
    # 这个dfs的作用就是返回经过当前节点的最大路径
    def dfs(self, root):

        if not root:
            return 0
        
        left_max = max(self.dfs(root.left), 0)
        right_max = max(self.dfs(root.right), 0)

        self.max_sum = max(self.max_sum, root.val + left_max + right_max)

        # 返回以改节点为起点或者终点的最大路径和
        return root.val + max(left_max, right_max)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容