力扣124. 二叉树中的最大路径和
给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
示例 1:
输入: [1,2,3]
1
/ \
2 3
输出: 6
示例 2:
输入: [-10,9,20,null,null,15,7]
-10
/ \
9 20
/ \
15 7
输出: 42
分析
这道题应该从叶子节点往上看,既然是找最大的路径和,而且不一定包含根节点,就分两种情况。
- 一个根节点和两个子节点组成了最大的。
- 这三个点不是最大的,往上继续找,就需要将当前节点的最大值更改为要么选左子树,要么选右子树。
以示例2举例,要么20、15、7 三个节点组成了最大的数值,要么不是最大数值,需要往上找,左子树最大,所以往上返回35。
代码
class Solution {
//给max赋值为最小值。
private int max = Integer.MIN_VALUE;
public int maxPathSum(TreeNode root) {
dfs(root);
return max;
}
public int dfs(TreeNode root){
//如果是null,返回0;
if(root == null) return 0;
//获取左右节点的值
int left = Math.max(dfs(root.left), 0);
int right = Math.max(dfs(root.right), 0);
//判断当前三个点是不是最大值。
max = Math.max(left+right+root.val, max);
//向上返回当前节点最大值
return Math.max(left+root.val, right+root.val);
}
}