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
  }
}
相关推荐
聚客AI15 小时前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
大怪v18 小时前
前端:人工智能?我也会啊!来个花活,😎😎😎“自动驾驶”整起!
前端·javascript·算法
惯导马工20 小时前
【论文导读】ORB-SLAM3:An Accurate Open-Source Library for Visual, Visual-Inertial and
深度学习·算法
骑自行车的码农21 小时前
【React用到的一些算法】游标和栈
算法·react.js
博笙困了1 天前
AcWing学习——双指针算法
c++·算法
moonlifesudo1 天前
322:零钱兑换(三种方法)
算法
NAGNIP2 天前
大模型框架性能优化策略:延迟、吞吐量与成本权衡
算法
美团技术团队2 天前
LongCat-Flash:如何使用 SGLang 部署美团 Agentic 模型
人工智能·算法
Fanxt_Ja2 天前
【LeetCode】算法详解#15 ---环形链表II
数据结构·算法·leetcode·链表
侃侃_天下2 天前
最终的信号类
开发语言·c++·算法