LC-124-二叉树中的最大路径和

Ordiiii

于 2023-02-23 21:01:47 发布

阅读量92

点赞数

分类专栏：算法 LeetCode-hot100 题解文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/Archerrrrr/article/details/129190075

版权

算法同时被 3 个专栏收录

79 篇文章 0 订阅

订阅专栏

题解

77 篇文章 1 订阅

订阅专栏

LeetCode-hot100

63 篇文章 0 订阅

订阅专栏

文章介绍了如何使用递归方法解决二叉树中找到经过某个根节点的最大路径和问题。通过维护一个以节点为端点的最大路径值的记录，避免重复计算，最终得到整棵树的最大路径和。代码示例展示了如何实现这个算法，并在预处理过程中更新结果。

摘要由CSDN通过智能技术生成

原题链接：二叉树中的最大路径和

个人解法

思路：

经过某个根的最长路径，我们可以抽象为下列情况的最大值

以左子树根为端点的最长路径
以右子树根为端点的最长路径
以左子树根为端点的最长路径+根结点的值+右子树根为端点的最长路径组成的路径长度

那么我们还需要求以某一个根我端点的最长路径，下列情况的最大值

以左子树根为端点的最长路径 + root->val
以右子树根为端点的最长路径 + root->val
root->val

故可以发现，上述就是递归定义，可以直接用递归来做，为了防止重复计算，可以记录以某根为端点的最长路径，如重复直接返回即可

时间复杂度： $O (n)$

代码：

/**
 * 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:
    unordered_map<TreeNode *, int> st;
    int res = -1010;

    int F(TreeNode *root) {
        if(root == nullptr) return 0;
        if(st.count(root)) return st[root];
        int t = max(F(root->left), F(root->right)) + root->val;
        t = max(t, root->val);
        st[root] = t;
        return t;
    }

    void pre(TreeNode *root) {
        if(root == nullptr) return;
        if(root->left) res = max(res, F(root->left));
        if(root->right) res = max(res, F(root->right));
        res = max(res, F(root));
        if(root->left && root->right) 
            res = max(res, F(root->left) + root->val + F(root->right));
        pre(root->left);
        pre(root->right);
    }

    int maxPathSum(TreeNode* root) {
        st[root] = F(root);
        pre(root);
        return res;
    }
};