Binary Tree Maximum Path Sum [leetcode] dp

最新推荐文章于 2025-04-24 10:46:30 发布

原创最新推荐文章于 2025-04-24 10:46:30 发布 · 793 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 专栏收录该内容

59 篇文章

订阅专栏

本文介绍了一种优化算法，通过节点属性计算单边和双边最大路径和，并使用递归和动态规划技巧实现路径求解，空间复杂度优化至O(1)，详细解释了算法逻辑及其实现细节。

a(i)：以节点i作为终点的单边最大路径和

b(i)：以节点i作为终点的双边边最大路径和

a(i) = max{ i->val, i->val + max{a(i->left), a(i->right) }};

b(i) = max{ i->val, i->val + max{a(i->left), a(i->right) } , i->val + a(i->left) + a(i->right)};

由于a(i), b(i)仅仅和a(i->left)和a(i->right) 有关，因此可以将空间压缩为O(1)

代码如下：

    int maxPathSum(TreeNode *root) {
        int res = INT_MIN;
        getSum(root, res);
        return res;
    }
    
    int getSum(TreeNode * root, int & res)
    {
        if (root == NULL) return 0;
        int l = getSum(root->left, res);
        int r = getSum(root->right, res);
        int a, b;
        a = max(root->val, root->val + max(l, r));//one side
        b = max(a, root->val + l + r);            //both side
        res = max(res, max(a, b));
        return a;
    }