题目
分析
当前节点的最大路径和取决于该节点的值与该节点的左右子节点的最大贡献值。
左右子节点的最大贡献值只有大于0时才会被计入当前节点的最大路径和。
当前节点的最大贡献值 等于 当前节点的值+左右节点最大贡献值中的较大值。
代码
/**
* Definition for a binary tree node.
* public 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 max=Integer.MIN_VALUE;//记录结果:最大路径和
public int maxPathSum(TreeNode root) {
dfs(root);
return max;
}
public int dfs(TreeNode root){
if(root==null)
return 0;
//左右子树只有贡献值>0时才计入
int left=Math.max(dfs(root.left),0);
int right=Math.max(dfs(root.right),0);
//更新结果
//节点的最大路径和取决于该节点的值与该节点的左右子节点的最大贡献值
max=Math.max(max,root.val+left+right);
//返回root的最大贡献值
return root.val+Math.max(left,right);
}
}
复杂度
时间复杂度O(N):其中 N 是二叉树中的节点个数。对每个节点访问不超过 2 次。
空间复杂度O(N):其中 N 是二叉树中的节点个数。空间复杂度主要取决于递归调用层数,最大层数等于二叉树的高度,最坏情况下,二叉树的高度等于二叉树中的节点个数。