题目描述
给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
输入输出
思路分析
二叉树后序遍历问题。节点的最大贡献值等于节点值与其子节点中的最大贡献值之和(对于叶节点而言,最大贡献值等于节点值)。空节点的最大贡献值等于 0。
节点的最大路径和取决于该节点的值与该节点的左右子节点的最大贡献值,如果子节点的最大贡献值为正,则计入该节点的最大路径和,否则不计入该节点的最大路径和。
维护一个全局变量 ans
存储最大路径和,在递归过程中更新 ans
的值,最后得到的ans
的值即为二叉树中的最大路径和。
- python实现:
- 子函数
class Solution:
ans=-sys.maxsize-1
def maxPathSum(self, root: TreeNode) -> int:
self.maxGain(root)
return self.ans
def maxGain(self,root):
if root==None:
return 0
left=max(0,self.maxGain(root.left))
right=max(0,self.maxGain(root.right))
self.ans=max(self.ans,left+right+root.val)
return max(left,right)+root.val
2.嵌套函数
class Solution:
ans=-sys.maxsize-1
def maxPathSum(self, root: TreeNode) -> int:
def maxGain(node):
if not node:
return 0
left=max(0,maxGain(node.left))
right=max(0,maxGain(node.right))
self.ans=max(self.ans,left+right+node.val)
return max(left,right)+node.val
maxGain(root)
return self.ans
C++实现:
class Solution {
public:
int ans=INT_MIN;//全局变量
int maxPathSum(TreeNode* root){
maxGain(root);
return ans;
}
int maxGain(TreeNode* root) {
if(root==NULL) return 0;
//递归计算左右节点最大贡献值
//只有在最大贡献值大于0时,才选取对应子节点
int left=max(0,maxGain(root->left));
int right=max(0,maxGain(root->right));
//更新答案
//节点的最大路径和取决于该节点的值与该节点的左右子节点的最大贡献值
ans=max(ans,left+right+root->val);
//返回节点的最大贡献值
return max(left,right)+root->val;
}
};