力扣 124. 二叉树中的最大路径和

103 阅读1分钟

🔗 leetcode.cn/problems/bi…

题目

  • 任一一条路径,求路径和最大的值

思路

  • 递归
  • 记录当前节点到叶子节点的最大路径和,等于 max(左孩子到叶子节点的最大路径和,右孩子到叶子节点的最大路径和) + node→val,或者 node→ val 本身
  • 最大路径和,则是这过程中,节点到叶子节点的路径和,以及左孩子最大路径和 + 右孩子最大路径和 + node→val 的最大值

代码

/**
 * 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 ans = 0;
    map<TreeNode*, int> max_route;
    void build(TreeNode* node) {
        if (node == nullptr) return;
        if (node->left == nullptr && node->right == nullptr) {
            max_route[node] = node->val;
            ans = max(ans, node->val);
            return;
        }
        build(node->left);
        build(node->right);
        max_route[node] = max(node->val, max(max_route[node->left], max_route[node->right]) + node->val);
        ans = max(ans, max_route[node]);
        ans = max(ans, max_route[node->left] + max_route[node->right] + node->val); 
    }
    int maxPathSum(TreeNode* root) {
        ans = root->val;
        max_route[nullptr] = 0;
        build(root);
        return ans;  
    }
};