leetcode题：124. 二叉树中的最大路径和（困难）

一、题目描述：124. 二叉树中的最大路径和（困难）

给定一个非空二叉树，返回其最大路径和。

本题中，路径被定义为一条从树中任意节点出发，达到任意节点的序列。该路径至少包含一个节点，且不一定经过根节点。

示例 1:

输入: [1,2,3]

       1
      / \
     2   3

输出: 6
示例 2:

输入: [-10,9,20,null,null,15,7]

   -10
   / \
  9  20
    /  \
   15   7

输出: 42

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-maximum-path-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、解题思路

递归查找左右子树的最大路径和，left，right，由于是寻找某个点到另外一个点的最大值。所以如果left或者right小于0的时候应该抛弃相加小于0的子树。此时应该left = max(left,0),right = max(right,0).然后判断最大值max_sum = max(max_sum , root->val + left,+right)。同时返回max(left+root->val,right+root->val)返回给上游。直到遍历完所有子树，统计所有节点加上左右子树的最大和，就是二叉树的最大路径和。

三、代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int maxPathSum(TreeNode* root) {
        if(root == NULL)
            return 0;
        int max_sum = INT_MIN;
        int root_max = maxPathSum(root,max_sum);
        return max_sum;
    }
    int maxPathSum(TreeNode* root,int & max_sum) {
        if(root == NULL)
            return 0;
        
        if(root->left == NULL && root->right == NULL)
        {
            if(root->val > max_sum)
            {
                max_sum = root->val;
            }
            return root->val;
        }
        int left = 0;
        int right = 0;
        if(root->left != NULL)
        {
            left = maxPathSum(root->left,max_sum);
        }
        if(root->right != NULL)
        {
            right = maxPathSum(root->right,max_sum);
        }
        if(left < 0)
            left = 0;
        if(right < 0)
            right = 0;
        int sum_left_right = left + right + root->val;
        //cout<<"root="<<root->val<<endl;
        //cout<<"sum_left_right="<<sum_left_right<<endl;
        if(sum_left_right > max_sum)
            max_sum = sum_left_right;
        if(root->left == NULL)
            return right + root->val;
        if(root->right == NULL)
            return left + root->val;
        int r_max = max(left,right);
        return r_max + root->val;         
    }
};