LeetCodes刷题总结6——二叉树中的最大路径和

题目

124. 二叉树中的最大路径和 - 力扣（LeetCode）

路径 被定义为一条从树中任意节点出发，沿父节点-子节点连接，达到任意节点的序列。同一个节点在一条路径序列中 至多出现一次 。该路径 至少包含一个 节点，且不一定经过根节点。

路径和 是路径中各节点值的总和。

给你一个二叉树的根节点 root ，返回其 最大路径和 。

示例 ：

 

输入：root = [-10,9,20,null,null,15,7]
输出：42
解释：最优路径是 15 -> 20 -> 7 ，路径和为 15 + 20 + 7 = 42

提示：

• 树中节点数目范围是 [1, 3 * 10^4]
• -1000 <= Node.val <= 1000

思路

本题难点在于题目中的路径和出发点未必是根节点，考虑以下一个基本单元。

 

有可能A+B+C是最大路径和，有可能只取A+B或A+C。因此我们需要定义一个全局变量记录最大路径和。同时对于某一分支结果为负，我们选择不加上那一段路径，可用max(0,分支路径)定义。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int res=INT_MIN; //全局变量记录最大路径和
    int help(TreeNode* root){
        if(root==nullptr) return 0; 
        int l=max(0,help(root->left)); //左边分支为负，不考虑添加这一分支
        int r=max(0,help(root->right)); //右边分支为负，不考虑添加这一分支
        int v=root->val+l+r; //基本单元中的A+B+C有可能为最大值
        res=max(res,v); //更新最大值
        return root->val+max(l,r); //返回A+B或者A+C
    }
    int maxPathSum(TreeNode* root) {
        help(root); //进入递归辅助函数
        return res;
    }
};