问题:
Given a binary tree, find the maximum path sum.
The path may start and end at any node in the tree.
For example:
Given the below binary tree,
1
/ \
2 3
Return6.
分析:
这道题是求树的路径和的题目,起始和终结于任何结点。在这里树没有被看成有向图,而是被当成无向图来寻找路径。因为这个路径的灵活性,我们需要对递归返回值进行一些调整,而不是通常的返回要求的结果。在这里,函数的返回值定义为以自己为根的一条从根到子结点的最长路径(这里路径就不是当成无向图了,必须往单方向走)。这个返回值是为了提供给它的父结点计算自身的最长路径用, 而结点自身的最长路径(也就是可以从左到右那种)则只需计算然后更新maxp 即可。
包含当前根节点的最大路径可能有以下情况:
第一种是只有自己的路径;
第二种是左子树的返回值加上当前节点;
第三种是右子树的返回值加上当前节点;
第四种是左右子树的路径加上当前节点(相当于一条横跨当前节点的路径)。
class Solution {
public:
int maxPathSum(TreeNode *root) {
if(root == nullptr)
return 0;
int maxp = -2147483648;
maxPath(root, maxp);
return maxp;
}
int maxPath(TreeNode *root, int& maxp){
if(root == nullptr)
return 0;
int left = max(0, maxPath(root->left, maxp));
int right = max(0, maxPath(root->right, maxp));
maxp = max(maxp, left+right+root->val);//因为left和right最小为0,所以此处求得四种情况的最大值与maxp比较
return root->val + max(left, right);
}
};