leetcode_124. 二叉树中的最大路径和

目录

一、题目内容

二、解题思路

三、代码


一、题目内容

给定一个非空二叉树,返回其最大路径和。

本题中,路径被定义为一条从树中任意节点出发,沿父节点-子节点连接,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。

示例 1:

输入:[1,2,3]

       1
      / \
     2   3

输出:6

示例 2:

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

   -10
   / \
  9  20
    /  \
   15   7

输出:42

二、解题思路

后序遍历,计算左右子树的路径和,如果存在小于0的数则不算在路径内(加0),每次记录一下根左右三个点的路径和,然后返回父节点选择左子树右子树的路径和,这样就可以记录子树选路径的情况。

三、代码

# Definition for a binary tree node.
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None
    def __repr__(self):
        return str(self.val)

class Solution:
    def maxPathSum(self, root: TreeNode) -> int:

        self.result = root.val
        def dfs(root):
            if not root:
                return 0
            left = max(dfs(root.left), 0)  # less than 0, no pass
            right = max(dfs(root.right), 0)  # less than 0, no pass

            self.result = max(root.val + left + right, self.result)
            return root.val + max(left, right)  # choose left or right
        dfs(root)
        return self.result


if __name__ == '__main__':
    # a = TreeNode(-10)
    # a.left = TreeNode(9)
    # a.right = TreeNode(20)
    # a.right.left = TreeNode(15)
    # a.right.right = TreeNode(7)
    a = TreeNode(1)
    a.left = TreeNode(-2)
    a.right = TreeNode(3)

    s = Solution()
    ans = s.maxPathSum(a)
    print(ans)
悲恋花丶无心之人 CSDN认证博客专家 深度学习 神经网络 Pytorch
计算机视觉在读研究生,熟悉Pytorch,MXNet,TensorFlow,Keras等深度学习框架,主要涉及的领域有目标检测,语义分割,超分辨率重建,行人重识别等。
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页
实付 9.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值