题目链接:牛客网
题目描述:
给定一个二叉树,请计算节点值之和最大的路径的节点值之和是多少。
这个路径的开始节点和结束节点可以是二叉树中的任意节点
例如:
给出以下的二叉树,
1↵ / ↵ 2 3
返回的结果为6
解题思路:
来自于Binary Tree Maximum Path Sum – LeetCode
这道题是求树的路径和的题目,不过和平常不同的是这里的路径不仅可以从根到某一个结点,而且路径可以从左子树某一个结点,然后到达右子树的结点,就像题目中所说的可以起始和终结于任何结点。在这里树没有被看成有向图,而是被当成无向图来寻找路径。因为这个路径的灵活性,我们需要对递归返回值进行一些调整,而不是通常的返回要求的结果。在这里,函数的返回值定义为以自己为根的一条从根到子结点的最长路径(这里路径就不是当成无向图了,必须往单方向走)。这个返回值是为了提供给它的父结点计算自身的最长路径用,而结点自身的最长路径(也就是可以从左到右那种)则只需计算然后更新即可。这样一来,一个结点自身的最长路径就是它的左子树返回值(如果大于0的话),加上右子树的返回值(如果大于0的话),再加上自己的值。而返回值则是自己的值加上左子树返回值,右子树返回值或者0(注意这里是“或者”,而不是“加上”,因为返回值只取一支的路径和)。在过程中求得当前最长路径时比较一下是不是目前最长的,如果是则更新。算法的本质还是一次树的遍历,所以复杂度是O(n)。而空间上仍然是栈大小O(logn)。
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int result=0;
int MaxNum(TreeNode *root) {
if(!root){
return 0;
}
int leftmax=0;
int rightmax=0;
//如果左边路径的最大和<0,则舍弃左边的路径返回0
if(root->left)
leftmax=max(0,MaxNum(root->left));
//如果右边路径的最大和<0,则舍弃左边的路径返回0
if(root->right)
rightmax=max(0,MaxNum(root->right));
//这个点可以构成的最大结果是root->val+leftmax+rightmax,如果小于result则不是最大路径,result不变,如果大于result,则是更长的路径,result更新
result=max(result,root->val+leftmax+rightmax);
//返回该节点到叶子结点路径上的存在的最大值
return max(0,max(root->val+leftmax,root->val+rightmax));
//只有当新节点的root->val+leftmax+rightmax>result才将新节点加入,但是返回的过程中max(root->val+leftmax,root->val+rightmax)节点子路径加节点的最大值总会被返回。这样就保证了一个节点本身是负的,在刚遍历到他时不会被加入,但是当遍历到他的父节点时值大于result,则更新。若仍然小,则继续舍弃。从而保证了路径的值始终是最大值。
}
int maxPathSum(TreeNode *root) {
if(!root){
return result;
}
result=INT_MIN;
MaxNum(root);
return result;
}
};