首页 > 代码库 > [LeetCode] Binary Tree Maximum Path Sum(最大路径和)

[LeetCode] 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.

题目意思很简单,就是给定一棵二叉树,求最大路径和。path 可以从任意 node 开始,到任意 node 结束。

这道题在 LeetCode 上的通过率只有 20% 多一点,并被标记为 Hard ,但实际上这是一道相当好的问题,在北美的 CS 求职面试中也是一道高频题,下文我将尝试用最容易理解的方式去分析这道题目。

我的思路

1. 我们直接从题目的问题来分析。给定一棵二叉树,我们怎么来求其最大路径和呢?稍做思考,不难得到以下结论:

当前二叉树的双路

最大路径和 = max(左子树的双路最大路径和, 右子树的双路最大路径和, 当前二叉树在包含根结点情况下的双路最大路径和);

2. 很显然,要解决上面的问题,关键在于怎么求二叉树在包含根结点情况下的双路最大路径和,而要求二叉树在包含根结点情况下的双路最大路径和,首先得先求二叉树的单路最大路径和,请看以下分析:

当前二叉树的单路

最大路径和 = max(左子树的单路最大路径和 + 根结点值, 右子树的单路最大路径和 + 根结点值, 根结点值);

当前二叉树在包含根结点情况下双路

最大路径和 = 左子树的单路最大路径和 + 根结点值 + 右子树的单路最大路径和;

3. 通过 1 和 2,我们很容易发现,该题可以采用分治法来解决,实际上,与二叉树有关的绝大多数问题,我们都可以采用分治的思想。通常,分治法避免使用全局变量,因此我将会封装一个 returnType 类型,来作为返回值,如下:

struct returnType{    returnType(const int maxSinglePathSum, const int maxDoublePathSum)              : maxSinglePathSum_(maxBranch), maxDoublePathSum_(maxSum)    { }        int maxSinglePathSum_;    int maxDoublePathSum_;};

4. 可以 AC 的完整源代码如下:

/** * Definition for binary tree * struct TreeNode { *     int val; *     TreeNode *left; *     TreeNode *right; *     TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */ struct returnType{    returnType(const int maxSinglePathSum, const int maxDoublePathSum)              : maxSinglePathSum_(maxSinglePathSum), maxDoublePathSum_(maxDoublePathSum)    { }        int maxSinglePathSum_;    int maxDoublePathSum_;};class Solution {public:    
    int maxPathSum(TreeNode *root)     {        return maxPathRec(root).maxDoublePathSum_;    }        returnType maxPathRec(TreeNode *root)    {        if (root == NULL)        {            return returnType(0, numeric_limits<int>::min());        }                // Devide        returnType left  = maxPathRec(root->left);        returnType right = maxPathRec(root->right);                // Conquer        int subMaxSinglePathSum = max(left.maxSinglePathSum_, right.maxSinglePathSum_);        int maxSinglePathSum    = root->val;        maxSinglePathSum        = max(subMaxSinglePathSum + maxSinglePathSum, maxSinglePathSum);                int subMaxDoublePathSum = max(left.maxDoublePathSum_, right.maxDoublePathSum_);        int maxDoublePathSum    = max(maxSinglePathSum, left.maxSinglePathSum_ + root->val + right.maxSinglePathSum_);        maxDoublePathSum        = max(subMaxDoublePathSum, maxDoublePathSum);                return returnType(maxSinglePathSum, maxDoublePathSum);    }};

[LeetCode] Binary Tree Maximum Path Sum(最大路径和)