方法:递归
首先,考虑实现一个简化的函数 maxGain(node),该函数计算二叉树中的一个节点的最大贡献值,具体而言,就是在以该节点为根节点的子树中寻找以该节点为起点的一条路径,使得该路径上的节点值之和最大。
具体而言,该函数的计算如下:
- 空节点的最大贡献值等于 00。
- 非空节点的最大贡献值等于节点值与其子节点中的最大贡献值之和(对于叶节点而言,最大贡献值等于节点值)。
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;
}
}
/*
二叉树最大路径和
myli25 2021.6.3
*/
public class MaxPathSum {
int maxvalue = Integer.MIN_VALUE;
public int maxPathSum(TreeNode root) {
maxGain(root);
return maxvalue;
}
public int maxGain(TreeNode node) {
if (node == null) {
return 0;
}
//递归计算左右节点的最大贡献值
//只有在最大贡献值大于0时,才会选取对应子节点
int leftGain = Math.max(maxGain(node.left), 0);
int rightGain = Math.max(maxGain(node.right), 0);
//节点的最大路径和取决于该节点的值与该节点的左右节点的最大贡献值
int priceNewpath = node.val + leftGain + rightGain;
//更新maxvalue
maxvalue = Math.max(maxvalue, priceNewpath);
//返回节点的最大贡献值
return node.val + Math.max(leftGain, rightGain);
}
}
复杂度分析
时间复杂度:O(N),其中 N 是二叉树中的节点个数。对每个节点访问不超过2 次。
空间复杂度:O(N),其中 N 是二叉树中的节点个数。空间复杂度主要取决于递归调用层数,最大层数等于二叉树的高度,最坏情况下,二叉树的高度等于二叉树中的节点个数。