二叉树中的最大路径和
路径: 被定义为一条从树中任意节点出发,沿父节点-子节点连接,达到任意节点的序列。同一个节点在一条路径序列中至多出现一次 。该路径 至少包含一个 节点,且不一定经过根节点。
路径和: 是路径中各节点值的总和。
题目要求:给你一个二叉树的根节点 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
题解:
class Solution {
int ans=-1000;
public int maxPathSum(TreeNode root) {
if(root==null) return 0;
oneSideMax(root);
return ans;
}
public int oneSideMax(TreeNode root){
if(root==null) return 0;
int left = Math.max(0,oneSideMax(root.left));
int right = Math.max(0,oneSideMax(root.right));
ans = Math.max(ans,left+right+root.val);
return Math.max(left,right)+root.val;
}
}
分析:
ans计算当前节点的最大路径和
oneSideMax则可以选择出两条边(两个子树)中有更大路径和的那一个,即从叶子选择1还是选择2走到目的节点。
思路:采用后序遍历的想法
先计算出左子树中已经选择出的最大一条边的路径和(oneSideMax),再接着是右子树,最后加上根的值(root.val)就是最大路径和。
比如示例二,按照我们的解题思路是:
自底向下,因为是后序遍历。
1.从9开始,因为它是叶子节点,容易算出其ans为9;return 0+9,代表从这里往上面走(-10)最大路径是9;
2.(-10)左子树遍历完,接着是右子树。
从15开始,和第一步类似,可知当前节点的路径和ans变成了15,所以最大值更新为15
3.从7开始,当前节点路径和为7比15小,不更新。
4.此时对于20,15,7这个子树来说,从左边只有一条 路15,右边也只有一条路7,所以这个树的最大路径和为15+7+20=42
5.到这里调用还没有结束,注意理解每一层次的调用,不要搞乱了顺序。
对于-10为根的这棵树,前面已经知道左边的路只有一条,现在要考虑右边走哪一条,此时就可以体会到oneSideMax的作用, 显而易见走20->15这一条,则完整路径是9->-10->20->15,其路径和却为34比42小,终得最大值是42.
对于整个过程,就像是在拼图一样,从最下面的叶子节点开始,一层一层往上拼,左右子树选择大的路径的边,比较一下子树的最大路径,慢慢更新,直到最终变成一个大树。