算法高频面试题：二叉树的最大路径和

最新推荐文章于 2021-08-11 16:36:42 发布

转载最新推荐文章于 2021-08-11 16:36:42 发布 · 425 阅读

CC 4.0 BY-SA版权

原文链接：https://leetcode-cn.com/problems/binary-tree-maximum-path-sum/solution/er-cha-shu-zhong-de-zui-da-lu-jing-he-by-leetcode-/

本文介绍了一个解决二叉树最大路径和问题的算法。通过递归计算左右子节点的最大贡献值，选择子节点的最大路径和，最终找到整棵树的最大路径和。此算法巧妙地运用了动态规划思想，实现了高效求解。

class Solution {
private:
    int maxSum = INT_MIN;

public:
    int maxGain(TreeNode* node) {
        if (node == nullptr) {
            return 0;
        }
        
        // 递归计算左右子节点的最大贡献值
        // 只有在最大贡献值大于 0 时，才会选取对应子节点
        int leftGain = max(maxGain(node->left), 0);
        int rightGain = max(maxGain(node->right), 0);

        // 节点的最大路径和取决于该节点的值与该节点的左右子节点的最大贡献值
        int priceNewpath = node->val + leftGain + rightGain;

        // 更新答案
        maxSum = max(maxSum, priceNewpath);

        // 返回节点的最大贡献值
        return node->val + max(leftGain, rightGain);
    }

    int maxPathSum(TreeNode* root) {
        maxGain(root);
        return maxSum;
    }
};