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

举报
悲恋花丶无心之人 发表于 2021/02/02 23:32:09 2021/02/02
【摘要】 目录 一、题目内容 二、解题思路 三、代码 一、题目内容 给定一个非空二叉树,返回其最大路径和。 本题中,路径被定义为一条从树中任意节点出发,沿父节点-子节点连接,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。 示例 1: 输入:[1,2,3]        1   ...

目录

一、题目内容

二、解题思路

三、代码


一、题目内容

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

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

示例 1:

输入:[1,2,3]

       1
      / \
     2   3

输出:6

示例 2:

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

   -10
   / \
  9  20
    /  \
   15   7

输出:42

二、解题思路

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

三、代码


  
  1. # Definition for a binary tree node.
  2. class TreeNode:
  3. def __init__(self, x):
  4. self.val = x
  5. self.left = None
  6. self.right = None
  7. def __repr__(self):
  8. return str(self.val)
  9. class Solution:
  10. def maxPathSum(self, root: TreeNode) -> int:
  11. self.result = root.val
  12. def dfs(root):
  13. if not root:
  14. return 0
  15. left = max(dfs(root.left), 0) # less than 0, no pass
  16. right = max(dfs(root.right), 0) # less than 0, no pass
  17. self.result = max(root.val + left + right, self.result)
  18. return root.val + max(left, right) # choose left or right
  19. dfs(root)
  20. return self.result
  21. if __name__ == '__main__':
  22. # a = TreeNode(-10)
  23. # a.left = TreeNode(9)
  24. # a.right = TreeNode(20)
  25. # a.right.left = TreeNode(15)
  26. # a.right.right = TreeNode(7)
  27. a = TreeNode(1)
  28. a.left = TreeNode(-2)
  29. a.right = TreeNode(3)
  30. s = Solution()
  31. ans = s.maxPathSum(a)
  32. print(ans)

文章来源: nickhuang1996.blog.csdn.net,作者:悲恋花丶无心之人,版权归原作者所有,如需转载,请联系作者。

原文链接:nickhuang1996.blog.csdn.net/article/details/108730702

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。