首页 > 代码库 > LeetCode: Binary Tree Maximum Path Sum [124]

LeetCode: Binary Tree Maximum Path Sum [124]

【题目】

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.



【题意】

给定一棵二叉树,找出其中路径和最大的路径,然会返回最大路径和。
本题中的路径不是从根节点到叶子节点这样的传统的路径,而是指的二叉树中任意两个节点之间的联通路径。


【思路】

    最直观的方法是找出任意两点之间的路径,然后选择和最大的路径。这种最笨的方法单单找两个节点之间的路径就要分4步走:
    1. 找到根到节点A的路径  O(logn)
    2. 找到根到节点B的路径  O(logn)
    3. 找出AB节点的最近公共祖先 O(logn)
    4. 计算途经最近公共祖先的路径和
    
    则搜索两两节点之间的路径的复杂就是O(n^2 logn), 显然是不可能接受的
    
    为此,我们换个角度来看。根据上面的分析,要找两个节点A,B之间的路径,我们需要先找两个节点的最近公共祖先R。那么实际上,只要A和B分别位于R的左右两棵子树上,两点之间的路径都需要经过节点R。那么针对根为R的这棵子树,我们可以计算这棵子树上经过R的最大路径。而二叉树上每个节点都都可视为一棵子树的根,所以我们的求解目标变为:计算出每个节点上以此节点为转折点的最大路径。
    那么这个最大路径怎么求呢? 
    我们假设要计算以R为转折点的最短路径,A和B分别是R的左孩子和右孩子,经过A和B的最大路径已知,如下:
    path_left(A), path_right(A)分别表示以A为转折点的最小路径在A的左右子树上的两段,则以A为转折点的最小路径即为path_left(A)->A->path_right(A)
    path_left(B), path_right(B)分别表示以B为转折点的最小路径在B的左右子树上的两段,则以B为转折点的最小路径即为path_left(B)->B->path_right(B)
    
    那么经过R的最短路径就可以表示为:max(path_left(A), path_right(B)) -> A -> R -> B -> max(path_left(B), path_right(B))
    
    所以,本题其实是一道典型的DP问题,上式即为迭代公式。
    


【代码】

/**
 * 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;
    }
};