二叉树中最大路径和

Leecode刷题

• 题目描述

给定一个非空二叉树，返回其最大路径和。
本题中，路径被定义为一条从树中任意节点出发，达到任意节点的序列。该路径至少包含一个节点，且不一定经过根节点。

• 示例
  输入: [1,2,3]

    1
   / \
  2   3
  

输出: 6

• 代码

/*最大值只有四种情况：节点， 节点及左树，节点及右树，左右树及节点。最后一种情况不能作为左右树继续递归，需要保存临时最大值*/
/**
 * 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 ans = INT_MIN;
        MaxSum(root,ans);
        return ans;
    }    
int MaxSum(TreeNode* root, int& ans)
{
        if(root == NULL)
            return 0;
        int left = max(0,MaxSum(root->left,ans));  
        int right = max(0,MaxSum(root->right,ans));
        ans = max(ans, left+right+root->val);
        return max(left+root->val,right+root->val);
    }
};