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

思路

问题描述:给定一个二叉树,每个节点包含一个整数值,路径被定义为节点序列,序列中每对相邻节点之间都存在一条边。求这棵二叉树中,路径和最大的路径的和。

解题思路:

这个问题可以通过递归来解决。对于每个节点,有以下几种情况:

  1. 最大路径和经过当前节点的左子树和右子树,并且向上连接:node.val + left_max + right_max
  2. 最大路径和仅经过当前节点的左子树,不连接到右子树:node.val + left_max
  3. 最大路径和仅经过当前节点的右子树,不连接到左子树:node.val + right_max
  4. 最大路径和仅包含当前节点,不连接到子树:node.val

其中,left_max 表示左子树的最大路径和,right_max 表示右子树的最大路径和。为了避免路径和为负的子路径对结果产生负影响,如果子路径和为负,我们选择不经过该子路径。

递归的基本情况是当节点为空时,路径和为0。递归过程中,我们可以计算每个节点的最大路径和,并在递归的同时更新全局最大路径和。

代码

scala 复制代码
class TreeNode(var _value: Int) {
  var value: Int = _value
  var left: TreeNode = null
  var right: TreeNode = null
}

object Solution {
  def maxPathSum(root: TreeNode): Int = {
    var maxSum = Int.MinValue

    def dfs(node: TreeNode): Int = {
      if (node == null) return 0
      
      // 递归计算左右子树的最大路径和,选择不经过负子路径
      val leftMax = Math.max(dfs(node.left), 0)
      val rightMax = Math.max(dfs(node.right), 0)
      
      // 更新全局最大路径和
      maxSum = Math.max(maxSum, node.value + leftMax + rightMax)
      
      // 返回以当前节点为终点的最大路径和
      node.value + Math.max(leftMax, rightMax)
    }

    dfs(root)
    maxSum
  }

  def main(args: Array[String]): Unit = {
    val root1 = new TreeNode(1)
    root1.left = new TreeNode(2)
    root1.right = new TreeNode(3)
    println(maxPathSum(root1))  // 输出 6

    val root2 = new TreeNode(-10)
    root2.left = new TreeNode(9)
    root2.right = new TreeNode(20)
    root2.right.left = new TreeNode(15)
    root2.right.right = new TreeNode(7)
    println(maxPathSum(root2))  // 输出 42
  }
}
相关推荐
慕容青峰21 分钟前
【蓝桥杯 2025 省 A 扫地机器人】题解
c++·算法·蓝桥杯·sublime text
_yingty_32 分钟前
GO语言入门:常用数学函数2
java·学习·算法·golang
猎猎长风36 分钟前
【数据结构和算法】3. 排序算法
数据结构·算法·排序算法
bookish_2010_prj1 小时前
链式栈和线性栈
数据结构·c++·算法
egoist20231 小时前
【C++指南】哈希驱动的封装:如何让unordered_map/set飞得更快更稳?【上】
数据结构·c++·算法·容器·哈希算法·散列表·c++11
Wang201220131 小时前
随机深林算法是分类还是回归?
算法·分类·回归
@蓝莓果粒茶2 小时前
LeetCode第158题_用Read4读取N个字符 II
前端·c++·python·算法·leetcode·职场和发展·c#
Heisenberg~2 小时前
C++回溯算法详解
开发语言·c++·算法
骑驴看星星a2 小时前
P10416 [蓝桥杯 2023 国 A] XYZ
算法·职场和发展·蓝桥杯
又过一个秋2 小时前
【sylar-webserver】重构日志系统
linux·c++·算法·重构