首页 > 其他 > 详细

【LeetCode】Binary Tree Maximum Path Sum

时间:2014-11-27 20:20:25      阅读:353      评论:0      收藏:0      [点我收藏+]

Binary Tree Maximum Path Sum

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

 

Return 6.

 

树结构显然用递归来解,解题关键:

1、对于每一层递归,只有包含此层树根节点的值才可以返回到上层。否则路径将不连续。

2、返回的值最多为根节点加上左右子树中的一个返回值,而不能加上两个返回值。否则路径将分叉。

在这两个前提下有个需要注意的问题,最上层返回的值并不一定是满足要求的最大值,

因为最大值对应的路径不一定包含root的值,可能存在于某个子树上。

因此解决方案为设置全局变量maxSum,在递归过程中不断更新最大值。

/**
 * 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 maxSum;
    Solution()
    {
        maxSum = INT_MIN;
    }
    int maxPathSum(TreeNode *root)
    {
        Helper(root);
        return maxSum;
    }
    int Helper(TreeNode *root) 
    {//return the maxSum containing root
     //if the maxSum without root is the true answer, update the maxSum directly
        if(root == NULL)
            return 0;
        else if(root->left == NULL && root->right == NULL)
        {
            //no child, return the root->val and maybe update the maxSum
            maxSum = (maxSum >= root->val) ? maxSum : root->val;
            return root->val;
        }
        else if(root->left != NULL && root->right == NULL)
        {
            int left = Helper(root->left);
            //return value must contain root->val
            //if left <= 0, drop it
            int tmp = root->val+max(0,left);
            maxSum = (maxSum >= tmp) ? maxSum : tmp;
            return tmp;
        }
        else if(root->left == NULL && root->right != NULL)
        {
            int right = Helper(root->right);
            //return value must contain root->val
            //if right <= 0, drop it
            int tmp = root->val+max(0,right);
            maxSum = (maxSum >= tmp) ? maxSum : tmp;
            return tmp;
        }
        else
        {
            int left = Helper(root->left);
            int right = Helper(root->right);
            //if left or right <= 0, drop it
            int tmp = root->val+max(0,left)+max(0,right);
            maxSum = (maxSum >= tmp) ? maxSum : tmp;
            //attention here! return value contain at most one child
            return root->val+max(0, max(left,right));
        }
    }
};

bubuko.com,布布扣

【LeetCode】Binary Tree Maximum Path Sum

原文:http://www.cnblogs.com/ganganloveu/p/4126953.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!