【重点】【DFS】124.二叉树中的最大路径和

Allenlzcoder

已于 2025-06-02 17:46:35 修改

阅读量397

点赞数 2

分类专栏：力扣Top100 文章标签：深度优先算法 DFS

于 2023-12-11 01:15:22 首次发布

本文链接：https://blog.youkuaiyun.com/Allenlzcoder/article/details/134917547

版权

力扣Top100 专栏收录该内容

100 篇文章

订阅专栏

本文介绍了如何使用递归方法，特别是深度优先搜索（DFS），在给定的二叉树中找到直径相同的最大路径和。Solution类中的`maxPathSum`和`dfs`方法分别用于计算单边分支最大和及整体最大和。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目
和求二叉树直径相同套路

Python

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:

    def __init__(self):
        self.res = -inf

    def maxPathSum(self, root: Optional[TreeNode]) -> int:
        if not root:
            return 0
        self.dfs(root)
        
        return self.res

    # 返回以root为根节点的最大路径和
    def dfs(self, root):
        if not root:
            return 0
        left_depth = max(self.dfs(root.left), 0)
        right_depth = max(self.dfs(root.right), 0)
        self.res = max(self.res, left_depth + right_depth + root.val)

        return root.val + max(left_depth, right_depth)

Java

class Solution {
    private int max = Integer.MIN_VALUE;
    public int maxPathSum(TreeNode root) {
        if (root == null) {
            return 0;
        }
        dfs(root);
        return max;
    }

    // 返回经过root的单边分支最大和
    public int dfs(TreeNode root) {
        if (root == null) {
            return 0;
        }
        int leftMax = Math.max(dfs(root.left), 0);
        int rightMax = Math.max(dfs(root.right), 0);
        max = Math.max(max, root.val + leftMax + rightMax);

        return root.val + Math.max(leftMax, rightMax);
    }
}