力扣(LeetCode)124. 二叉树中的最大路径和(C++)

清墨韵染

已于 2022-11-30 12:11:10 修改

阅读量603

点赞数 2

分类专栏：墨染leetcode 文章标签： leetcode c++ 深度优先递归算法

于 2022-11-30 12:05:38 首次发布

本文链接：https://blog.csdn.net/Innocence02/article/details/128111706

版权

墨染leetcode 专栏收录该内容

138 篇文章 2 订阅

订阅专栏

深度优先遍历

自底向上的递归。

对于二叉树，我们令每个结点作为 $L C A$ (最近公共祖先)，
结点 $u$ 作为 $L C A$ ，经过它的最大路径 $=$ 往左子树的最大路径 $+$ 往右子树的最大路径 $+$ 它自己的路径。
有 $ans = max(ans,v_u+l+r$ )

对于结点 $u$ 的上层结点，经过 $u$ 的最大路径 $= m a x ($ 往左走 $,$ 往右走 $) +$ 它自己的路径
有 $path_u=max(l,r)+v_u$

递归时，自底向上，记录结点的左右最大路径 $l / r$ ，维护结点作为 $L C A$ 的 $a n s$ 。遇到空结点，没有路径，即为 $0$ 。

class Solution {
public:
    int ans;
    int maxPathSum(TreeNode* root) {
        ans = INT_MIN;
        dfs(root);
        return ans;
    }
    int dfs(TreeNode *root){
        if(!root) return 0;
        int l = max(0,dfs(root->left)),r = max(0,dfs(root->right));
        ans = max(ans,l+r+root->val);
        return root->val + max(l,r);
    }
};

时间复杂度 : $O (n)$ ，每个结点最多遍历一次，时间复杂度 $O (n)$ 。
空间复杂度 : $O (∣ h ∣)$ ，函数压栈的最大深度 $O (n)$ 。