原题链接:. - 力扣(LeetCode)
二叉树中的 路径 被定义为一条节点序列,序列中每对相邻节点之间都存在一条边。同一个节点在一条路径序列中 至多出现一次 。该路径 至少包含一个 节点,且不一定经过根节点。
路径和 是路径中各节点值的总和。
给你一个二叉树的根节点 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 *
]
-1000 <= Node.val <= 1000
思路:
对于每个节点,以这个节点为根节点的路径最大和 为 该节点的节点值 + 左右子节点的最大贡献值。所以只需要遍历整棵二叉树,对于每一个节点:1.计算其最大贡献值 并返回2.计算以该节点为根节点的最大路径和 curSum 并更新全局最大路径和 maxSum。
节点的最大贡献值分以下两种情况:
1.节点为空,其最大贡献值为0。
2.节点为叶子节点,则其最大贡献值为叶子节点的节点值。(节点不为叶子节点,最大贡献值为节点值 + 左右子节点中贡献值较大者。)
代码:
class Solution {
int maxSum = Integer.MIN_VALUE;
public int maxPathSum(TreeNode root) {
maxGain(root);
return maxSum;
}
public int maxGain(TreeNode root){
if(root==null){
return 0;
}
//只有当左右子节点的最大贡献为正数时,才取左右子节点的贡献值
//用于计算当前节点的路径和
int leftGain = Math.max(maxGain(root.left),0);
int rightGain = Math.max(maxGain(root.right),0);
int curSum = root.val + leftGain + rightGain;
maxSum = Math.max(maxSum,curSum);
//返回当前节点的最大贡献值
return root.val + Math.max(leftGain,rightGain);
}
}