LeetCode124二叉树中的最大路径和

时间:2020-07-27 17:57:43   收藏:0   阅读:83

题目链接

https://leetcode-cn.com/problems/binary-tree-maximum-path-sum/

题解

// Problem: LeetCode 124
// URL: https://leetcode-cn.com/problems/binary-tree-maximum-path-sum/
// Tags: Tree Recursion DFS
// Difficulty: Hard

#include <iostream>
#include <algorithm>
using namespace std;

struct TreeNode
{
    int val;
    TreeNode *left;
    TreeNode *right;
};

class Solution
{
private:
    int maxSum = INT_MIN; // 全局最大路径和

    // 功能:求一颗树的最大路径和,要求从根节点开始,在左子树或右子树结束,也可不经过左子树和右子树,即至多经过一个子树
    int dfs(TreeNode *root)
    {
        if (root == nullptr)
            return 0;
        int left = dfs(root->left);
        int right = dfs(root->right);
        // 过程中更新全局最大路径和:根节点必须经过,左右子树可经过可不经过
        maxSum = max(maxSum, root->val + max(left, 0) + max(right, 0));
        // 从根节点开始,在左子树或右子树结束,也可不经过左子树和右子树,即至多经过一个子树
        return max(root->val, root->val + max(max(left, right), 0));
    }

public:
    // 求一颗树的最大路径和,不要求经过根节点,起始和终止节点任意
    int maxPathSum(TreeNode *root)
    {
        dfs(root);
        return this->maxSum;
    }
};

作者:@臭咸鱼

转载请注明出处:https://www.cnblogs.com/chouxianyu/

欢迎讨论和交流!


原文:https://www.cnblogs.com/chouxianyu/p/13386394.html

评论(0
© 2014 bubuko.com 版权所有 - 联系我们:wmxa8@hotmail.com
打开技术之扣,分享程序人生!