LeetCode.H124.二叉树中的最大路径和

最新推荐文章于 2024-10-12 23:06:01 发布

cute-ghost

最新推荐文章于 2024-10-12 23:06:01 发布

阅读量200

点赞数

分类专栏： LEETCODE # HOT100 文章标签： leetcode 算法

本文链接：https://blog.csdn.net/a695484357/article/details/126678459

版权

LEETCODE 同时被 2 个专栏收录

37 篇文章 0 订阅

订阅专栏

HOT100

22 篇文章 0 订阅

订阅专栏

LeetCode.H124.二叉树中的最大路径和

题目：

在这里插入图片描述

题目大意：

求树中的最大路径。

数据范围：

如图所示

思路：

思路与LeetCode.M687.最长同值路径大致一致，都是递归的处理一个节点 node 的左右孩子节点。

对于一个节点 node1，我们可知以 node1 为根的树的最大路径，可能是由以下几种情况组成：

1. 以 node1 为出发点，向 node1.left 延伸出的路径。
1. 以 node1 为出发点，向 node1.right 延伸的路径。
1. node1 在中间，向 node1.left 和 node1.right 延伸的路径。
1. 最大路径在以 node1.left 或 node1.right 为根节点的树中。

这样的话，我们可以递归的处理每一个节点，求情况1.2.3.的最大值来更新 ans，以全局变量 ans 记录最大路径的值。

现在思考以节点 node1 延伸出去的左右孩子节点的路径值（包括 node1 ），也就是函数的返回值怎么计算。节点 node1 向其左孩子 node2 延伸出的路径为 l ，向其右孩子 node3 延伸出的路径为 r，则可能有以下几种情况组成：

1. 只有 node1 节点。
1. 有 node1 节点和 node2 节点（的路径值）。
1. 有 node1 节点和 node3 节点（的路径值）。

我们取这三种情况的最大值则为 node1 节点的路径值，并返回作为 node1 的父节点的 l （或r）使用。

代码：

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode() {}
    TreeNode(int val) { this.val = val; }
    TreeNode(int val, TreeNode left, TreeNode right) {
        this.val = val;
        this.left = left;
        this.right = right;
    }
}

class Solution {
    int ans = -(int)1e9;
    public int getAns(TreeNode root){
        if (root == null)
            return 0;
        int l = getAns(root.left), r = getAns(root.right);

        int t = Math.max(l, r) + root.val;
        t = Math.max(t ,root.val);
        t = Math.max(t, root.val + l + r);
        ans = Math.max(ans, t);

        int temp = Math.max(root.val, root.val + l);
        temp = Math.max(temp, root.val + r);
        return temp;
    }

    public int maxPathSum(TreeNode root) {
        getAns(root);
        return ans;
    }
}

public class Main {
    public static void main(String[] args) {
		
    }
}