码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode -124. Binary Tree Maximum Path Sum

时间:2016-08-28 01:20:29      阅读:140      评论:0      收藏:0      [点我收藏+]

标签:

Given a binary tree, find the maximum path sum.

For this problem, a path is defined as any sequence of nodes from some starting node to any node in the tree along the parent-child connections. The path does not need to go through the root.

For example:
Given the below binary tree,

       1
      /      2   3

 

Return 6.

求二叉树的最大路径和,路径可以从任一节点开始到任一节点结束。要注意节点值可以为负。

这题可以用分治法递归实现。我们可以把问题拆分开:最后的最大路径必然有一个“最高点”,因此我们只要针对所有结点,求出:如果路径把这个节点作为“最高点”,路径最长可达多少?记为max,然后在max中求出最大值MAX即为所求结果。

代码如下:

/**
 * 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) {
        rec(root);
        return max;
    }
    int rec(TreeNode *root)
    {
        if(root == NULL)
            return 0;
        int l = rec(root->left);
        int r = rec(root->right);
        if(root->val>max)   max = root->val;
        if(root->val+l>max) max = root->val+l;
        if(root->val+r>max) max = root->val+r;
        if(root->val+l+r > max) max = root->val+l+r;
        int maxReturn = root->val>(root->val+l) ? root->val:root->val+l;
        return (root->val+r)>maxReturn ? root->val+r : maxReturn;
    }
private:
    int max = INT_MIN;
};

rec的返回值为max(root->val,root->val+r,root->val+l)。这是因为要和上层结点连成一条路径。

 

leetcode -124. Binary Tree Maximum Path Sum

标签:

原文地址:http://www.cnblogs.com/tonychen-tobeTopCoder/p/5814155.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!