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
  }
}
相关推荐
DebugKitty25 分钟前
C语言14-指针4-二维数组传参、指针数组传参、viod*指针
c语言·开发语言·算法·指针传参·void指针·数组指针传参
qystca30 分钟前
MC0241防火墙
算法
jghhh011 小时前
Arduino声控RGB矩阵音乐节奏灯DIY全攻略
线性代数·矩阵
行然梦实3 小时前
粒子群优化算法(Particle Swarm Optimization, PSO) 求解二维 Rastrigin 函数最小值问题
算法·机器学习·数学建模
XH华3 小时前
C语言第六章函数递归
c语言·开发语言·算法
斯安3 小时前
LRU(Least Recently Used)原理及算法实现
算法
go54631584654 小时前
基于LSTM和GRU的上海空气质量预测研究
图像处理·人工智能·深度学习·神经网络·算法·gru·lstm
亮亮爱刷题4 小时前
算法提升之数论(矩阵+快速幂)
线性代数·算法·矩阵
亿坊电商4 小时前
AI 数字人在处理音频时,如何确保声音的自然度?
人工智能·算法·音视频
向左转, 向右走ˉ4 小时前
随机森林算法原理及优缺点
算法·随机森林·机器学习