LeetCode: Binary Tree Maximum Path Sum [124]

/**
 * 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 maxPath(int& maxSum, TreeNode *root){
        if(root==NULL)return 0;
        
        //计算来自左子树的最大路径和
        int maxLeft=maxPath(maxSum, root->left);
        //计算来自右子树的最大路径和
        int maxRight=maxPath(maxSum, root->right);
        //计算以当前节点为转折的最大路径和
        int curMaxSum=root->val;
        if(maxLeft>0)curMaxSum+=maxLeft;    //如果从左右孩子过来的最大和是负值，则连接到当前节点，负值会使合并后的路径和更小，因此只在路径和为正的时候再连接到当前节点，具体表现为路径和相加。
        if(maxRight>0)curMaxSum+=maxRight;
        //更新结果
        if(curMaxSum>maxSum)maxSum=curMaxSum;
        
        //同样需要考虑maxLeft和maxRight的正负
        int largerSum = max(maxLeft, maxRight);
        if(largerSum>0)return root->val+largerSum;
        else return root->val;
    }

    int maxPathSum(TreeNode *root) {
        if(root==NULL)return 0;
        int max=INT_MIN;
        maxPath(max, root);
        return max;
    }
};