给出一棵二叉树,寻找一条路径使其路径和最大,路径可以在任一节点中开始和结束(路径和为两个节点之间所在路径上的节点权值之和)。
最大的路径和有可能是哪些情况。
- 第一种是左子树的路径加上当前节点
- 第二种是右子树的路径加上当前节点
- 第三种是左右子树的路径加上当前节点(相当于一条横跨当前节点的路径)
- 第四种是只有自己的路径。
这四种情况只是用来计算以当前节点根的最大路径。
如果当前节点上面还有节点,那它的父节点是不能累加第三种情况的。所以我们要计算两个最大值,
一个是当前节点下的最大路径和,
一个是如果要连接父节点时最大的路径和
我们用前者更新全局最大量,用后者返回递归值就行了。
代码
递归函数的返回值是:当前这个结点的左子树最大路径和,或者是右子树最大路径和。再加上结点值。因为,这样得到的一条最大路径,就可以作为这个结点的父节点的一条路径,用于父节点的计算。
class Solution {
public:
int max_res;
int maxPathSum(TreeNode *root) {
max_res=INT_MIN;
maxSum(root);
return max_res;
}
int maxSum(TreeNode *root)
{
if(root==NULL)
return 0;
int leftMaxPath=max(0,maxSum(root->left));
int rightMaxPath=max(0,maxSum(root->right));
max_res=max(max_res,root->val+leftMaxPath+rightMaxPath);
return max(leftMaxPath,rightMaxPath)+(root->val);
}
};