给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 1:
输入: [1,2,3]
1
/ \
2 3
输出: 6 示例 2:
输入: [-10,9,20,null,null,15,7]
-10
/ \
9 20
/ \
15 7
输出: 42
这道题是一个post-order-traversal的变形题目。我们很容易想到left的最大路径和right的最大路径求完之后更新最终结果的状态。这时候我们就会去思考递归子问题的代码怎么去构造。会发现面临两个关键问题:
return Math.max(leftSum, rightSum) + root.val;
但是我们的递归只能返回一个参数啊。怎么办呢?
回头再看看我们的第1点,return 回去是为了父节点的下一次计算;而第2点,不经过根节点的最大值,我们只需要记录下来即可,不需要return ,因为它并不涉及后面的计算。
class Solution {
private int maxSum;
public int maxPathSum(TreeNode root) {
maxSum = Integer.MIN_VALUE;
helper(root);
return maxSum;
}
private int helper(TreeNode root) {
// [-1, -2, -3]
if (root == null) return 0;
int leftSum = Math.max(helper(root.left), 0);
// 和0比较要么要这个分支,要么不要这个分支
int rightSum = Math.max(helper(root.right), 0);
// 当前节点路径下最大值,对应解析中的第2点
maxSum = Math.max(leftSum + rightSum + root.val, maxSum);
// 这个分支最大的值是多少
return Math.max(leftSum, rightSum) + root.val;
}
}
本文分享自 Leetcode名企之路 微信公众号,前往查看
如有侵权,请联系 cloudcommunity@tencent.com 删除。
本文参与 腾讯云自媒体同步曝光计划 ,欢迎热爱写作的你一起参与!